🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal models, for both inference and training.

Tensors and Dynamic neural networks in Python with strong GPU acceleration

Unsloth Studio is a web UI for training and running open models like Qwen, DeepSeek, gpt-oss and Gemma locally.

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

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

The fundamental package for scientific computing with Python.

⚡ A Fast, Extensible Progress Bar for Python and CLI

SGLang is a high-performance serving framework for large language models and multimodal models.

A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training

Fast and memory-efficient exact attention

🤗 The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools

Mamba SSM architecture

Wan: Open and Advanced Large-Scale Video Generative Models

Wan: Open and Advanced Large-Scale Video Generative Models

RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable). We are at RWKV-7 "Goose". So it's combining the best of RN…

Open source code for AlphaFold 2.

Nano vLLM

Implementation of Denoising Diffusion Probabilistic Model in Pytorch

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

Accessible large language models via k-bit quantization for PyTorch.

A concise but complete full-attention transformer with a set of promising experimental features from various papers

Open-source unified multimodal model

A PyTorch native platform for training generative AI models

Denoising Diffusion Probabilistic Models

A fast AI Video Generator for the GPU Poor. Supports Wan 2.1/2.2, Qwen Image, Hunyuan Video, LTX Video and Flux.

TFDS is a collection of datasets ready to use with TensorFlow, Jax, ...

The best OSS video generation models, created by Genmo

Entropy Based Sampling and Parallel CoT Decoding