🤗 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.

Collection of awesome LLM apps with AI Agents and RAG using OpenAI, Anthropic, Gemini and opensource models.

Deep Learning for humans

Universal memory layer for AI Agents; Announcing OpenMemory MCP - local and secure memory management.

A collection of design patterns/idioms in Python

Ray is an AI compute engine. Ray consists of a core distributed runtime and a set of AI Libraries for accelerating ML workloads.

A community-maintained Python framework for creating mathematical animations.

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

Let us control diffusion models!

Chat with your documents on your local device using GPT models. No data leaves your device and 100% private.

PyTorch implementations of Generative Adversarial Networks.

OpenAI Baselines: high-quality implementations of reinforcement learning algorithms

Mamba SSM architecture

Machine learning, in numpy

Train transformer language models with reinforcement learning.

🦉 Data Versioning and ML Experiments

An open-source, code-first Python toolkit for building, evaluating, and deploying sophisticated AI agents with flexibility and control.

Low-code framework for building custom LLMs, neural networks, and other AI models

An educational resource to help anyone learn deep reinforcement learning.

Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch

Always know what to expect from your data.

PyTorch package for the discrete VAE used for DALL·E.

A framework for training and evaluating AI models on a variety of openly available dialogue datasets.

Hydra is a framework for elegantly configuring complex applications

A logical, reasonably standardized, but flexible project structure for doing and sharing data science work.

An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.

StarCraft II Learning Environment

Official PyTorch Implementation of "Scalable Diffusion Models with Transformers"

BertViz: Visualize Attention in NLP Models (BERT, GPT2, BART, etc.)