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

Simple text to phones converter for multiple languages

Basic scripts for interacting with the Stability AI Stable Diffusion API.

Karras et al. (2022) diffusion models for PyTorch

🐍 Geometric Computer Vision Library for Spatial AI

🔍 🐍 Like pstack but for Python!

Dias: Dynamic Rewriting of Pandas Code

🎹🎸A music theory library with a command-line interface

Easily turn large sets of image urls to an image dataset. Can download, resize and package 100M urls in 20h on one machine.

FFCV: Fast Forward Computer Vision (and other ML workloads!)

A library to analyze PyTorch traces.

Awesome utilities for performance profiling

A runtime shape checker and auto-annotator for tensor programs (pronounced "stanley")

An implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed libraries

Containers for machine learning

Recommendations at "Reasonable Scale": joining dataOps with recSys through dbt, Merlin and Metaflow

A database that operates at CLOUD SCALE

A Point Process Toolbox Based on PyTorch

XLS: Accelerated HW Synthesis

FTPipe and related pipeline model parallelism research.

Training and serving large-scale neural networks with auto parallelization.

Convert a mono input stream to a stereo one

Converting MIDI to blender NLA for music-generated animation

automatic video supercuts with python

PyTorch extensions for high performance and large scale training.

A tensor-aware point-to-point communication primitive for machine learning

Multithreaded Python without the GIL

A collection of (public) notes on assorted topics