Tensors and Dynamic neural networks in Python with strong GPU acceleration

Open standard for machine learning interoperability

PyTorch Extension Library of Optimized Graph Cluster Algorithms

Vim syntax file for apple metal shading language

Multi-language library for the calculation of spherical harmonics in Cartesian coordinates

[MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.

Batch computation of the linear assignment problem on GPU.

D2Go is a toolkit for efficient deep learning

OpenMMLab Computer Vision Foundation

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

Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.

oneAPI Deep Neural Network Library (oneDNN)

Inference Llama 2 with a model compiled to native code by TorchInductor

Action for checking out a repo

DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it

Cryptographically secure pseudorandom number generators for PyTorch

docTR (Document Text Recognition) - a seamless, high-performing & accessible library for OCR-related tasks powered by Deep Learning.

[ICLR 2020] Once for All: Train One Network and Specialize it for Efficient Deployment

CUDA Templates for Linear Algebra Subroutines

Development repository for the Triton language and compiler

Making big AI models cheaper, easier, and more scalable