facebookexperimental/triton

Commits: 63

facebookincubator/AITemplate

AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (NVIDIA GPU) and MatrixCore (AMD GPU) inference.

Commits: 27

uwsampl/calyx

Commits: 20

cucapra/calyx

Commits: 20

calyxir/calyx

Commits: 20

pytorch-labs/tritonbench

Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.

Commits: 17

cucapra/dahlia-evaluation

Commits: 14

sampsyo/cs6120

advanced compilers

Commits: 14

uwsampl/dahlia

Commits: 9

meta-pytorch/torchrec

Commits: 6

pytorch/pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Commits: 6

pytorch/torchrec

Pytorch domain library for recommendation systems

Commits: 5

pytorch/FBGEMM

FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/

Commits: 4

rrnewton/reverie

Commits: 1

pytorch/TensorRT

PyTorch/TorchScript/FX compiler for NVIDIA GPUs using TensorRT

Commits: 1

pytorch/benchmark

TorchBench is a collection of open source benchmarks used to evaluate PyTorch performance.

Commits: 1

siriusinftra/pytorch

Commits: 1

medav/torchbench-ncu

Commits: 1

sampsyo/bril

an educational compiler intermediate representation

Commits: 1