PyTorch bindings for CUTLASS grouped GEMM.

🚀 A simple way to train and use PyTorch models with multi-GPU, TPU, mixed-precision

LLM training code for MosaicML foundation models

A Data Streaming Library for Efficient Neural Network Training

Pipeline is an open source API for building AI/ML workflows

ML powered analytics engine for outlier detection and root cause analysis.

Research and experimental code related to Opacus, an open-source library for training PyTorch models with Differential Privacy

Code repo for the paper: "SubMix: Practical Private Prediction for Large-scale Language Models"

ColBERT: Contextualized Late Interaction over BERT (SIGIR'20)

The world's most expensive version of Conway's Game of Life - running on the Ethereum Blockchain

An implementation of a deep learning recommendation model (DLRM)