Official repository for the Boltz biomolecular interaction models

CUDA checkpoint and restore utility

Official implementation of All Atom Diffusion Transformers (ICML 2025)

Optimizing AlphaFold Training and Inference on GPU Clusters

A bidirectional pipeline parallelism algorithm for computation-communication overlap in DeepSeek V3/R1 training.

Cost-efficient and pluggable Infrastructure components for GenAI inference

Production-ready, Light, Flexible and Extensible ASGI API framework | Effortlessly Build Performant APIs

An extremely fast Python package and project manager, written in Rust.

Cuda extensions for PyTorch

pathlib api extended to use fsspec backends

Improved build system generator for CPython C, C++, Cython and Fortran extensions

TensorDict is a pytorch dedicated tensor container.

Audiocraft is a library for audio processing and generation with deep learning. It features the state-of-the-art EnCodec audio compressor / tokenizer, along with MusicGen, a simple and controllable…

Unified-Modal Speech-Text Pre-Training for Spoken Language Processing

Integrate cutting-edge LLM technology quickly and easily into your apps

JAX implementation of OpenAI's Whisper model for up to 70x speed-up on TPU.

General technology for enabling AI capabilities w/ LLMs and MLLMs

A playbook for systematically maximizing the performance of deep learning models.

Data Efficient Decision Making

A Non-Autoregressive Text-to-Speech (NAR-TTS) framework, including official PyTorch implementation of PortaSpeech (NeurIPS 2021) and DiffSpeech (AAAI 2022)

A repository for benchmarking neural vocoders by their quality and speed.

A python package to analyze and compare voices with deep learning

Clone a voice in 5 seconds to generate arbitrary speech in real-time

ipapy is a Python module to work with International Phonetic Alphabet (IPA) strings

An implementation of the Debug Adapter Protocol for Python

Charsiu: A neural phonetic aligner.

Library for 8-bit optimizers and quantization routines.

A high-performance Python-based I/O system for large (and small) deep learning problems, with strong support for PyTorch.

This is an open-source implementation of the ITU P.808 standard for "Subjective evaluation of speech quality with a crowdsourcing approach" (see https://www.itu.int/rec/T-REC-P.808/en). It uses Ama…