FB (Facebook) + GEMM (General Matrix-Matrix Multiplication) - https://code.fb.com/ml-applications/fbgemm/
Collection of common code that's shared among different research projects in FAIR computer vision team.
Tensors and Dynamic neural networks in Python with strong GPU acceleration
A deep learning library for video understanding research.
Simple (fast) transformer inference in PyTorch with torch.compile + lit-llama code