Tensors and Dynamic neural networks in Python with strong GPU acceleration

Pytorch domain library for recommendation systems

End-to-end solution for enabling on-device AI across mobile and edge devices for PyTorch models

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

Enabling PyTorch on XLA Devices (e.g. Google TPU)

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

PyTorch tutorials.

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.

Detectron2 is a platform for object detection, segmentation and other visual recognition tasks.

Mobile vision models and code

An implementation of a deep learning recommendation model (DLRM)

✨ Monorepo for all the tooling which enables ESLint to support TypeScript

VSCode Extension for C0. Used by 15-122 students at Carnegie Mellon University.

Free Open Source Document Management System

Collection of generative models in Pytorch version.