DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.

Bayesian optimization in PyTorch

Dummy repo

Open source implementation of AlphaFold3

LLaMA-Omni is a low-latency and high-quality end-to-end speech interaction model built upon Llama-3.1-8B-Instruct, aiming to achieve speech capabilities at the GPT-4o level.

Deep universal probabilistic programming with Python and PyTorch

Spot Sim2Real Infrastructure

The lightweight PyTorch wrapper for high-performance AI research. Scale your models, not the boilerplate.

A library for efficient similarity search and clustering of dense vectors.

The repository provides code for running inference with the Meta Segment Anything Model 2 (SAM 2), links for downloading the trained model checkpoints, and example notebooks that show how to use th…

Qiskit is an open-source SDK for working with quantum computers at the level of extended quantum circuits, operators, and primitives.

Hackable and optimized Transformers building blocks, supporting a composable construction.

FAIR Sequence Modeling Toolkit 2

Train transformer language models with reinforcement learning.

Diffusion models in PyTorch

OpenAI Baselines: high-quality implementations of reinforcement learning algorithms

Utilities intended for use with Llama models.

Experimental design and (multi-objective) bayesian optimization.

🤗 LeRobot: End-to-end Learning for Real-World Robotics in Pytorch

Foundational Models for State-of-the-Art Speech and Text Translation

Focus on prompting and generating

The simplest, fastest repository for training/finetuning medium-sized GPTs.

Native PyTorch library for quantization and sparsity

A concrete syntax tree parser and serializer library for Python that preserves many aspects of Python's abstract syntax tree

The Incredible PyTorch: a curated list of tutorials, papers, projects, communities and more relating to PyTorch.

A curated awesome list of flake8 extensions. Feel free to contribute! 🎓

A comprehensive list of pytorch related content on github,such as different models,implementations,helper libraries,tutorials etc.

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

NeMo: a toolkit for conversational AI