ezyang

htmlpurifier

Standards compliant HTML filter written in PHP

ghstack

Submit stacked diffs to GitHub on the command line

convolution-visualizer

Convolution visualizations

compact

Compact regions library for Haskell

expecttest

torchdbg

PyTorch centric eager mode debugger

ghc-shake

ghc --make reimplemented with Shake

pytorch-unattached

Tensors and Dynamic neural networks in Python with strong GPU acceleration

tlparse

TORCH_LOGS parser for PT2

cabal

Official upstream development repository for Cabal and cabal-install

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

compiler-explorer

Run compilers interactively from your web browser and interact with the assembly

labeler-github-action

Labeling GitHub action for PyTorch issues/pull requests

accelerated-pytorch-transformers-generation

LLM-Training-Puzzles

What would you do with 1000 H100s...

pdfs

algorithmic-efficiency

MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvements in both training algorithms and models.

benchmark

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

detectron2

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

FBGEMM

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

lit-llama

Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Adapter fine-tuning, pre-training. Apache 2.0-licensed.

minGPT

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

pytorch_sparse

PyTorch Extension Library of Optimized Autograd Sparse Matrix Operations

recipes

Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research techniques without significant engineering overhead.Specifically, recipes aims to provide- Consistent access to pre-trained SOTA models ready for production- Reference implementations for SOTA research reproducib

testing-ideal-computing-machine

Testing repo

torchrec

Pytorch domain library for recommendation systems

transformers

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

triton

Development repository for the Triton language and compiler

triton-llvm-releases

vision

Datasets, Transforms and Models specific to Computer Vision