State-of-the-art diffusion models for image and audio generation
LLM training code for MosaicML foundation models
A library for accelerating Transformer models on NVIDIA GPUs
Open platform for training, serving, and evaluating language models
High quality, fast, modular reference implementation of SSD in PyTorch