Model parallel transformers in JAX and Haiku
☆6,363Jan 21, 2023Updated 3 years ago
Alternatives and similar repositories for mesh-transformer-jax
Users that are interested in mesh-transformer-jax are comparing it to the libraries listed below
Sorting:
- An implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed libraries☆7,396Feb 3, 2026Updated last month
- An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.☆8,287Feb 25, 2022Updated 4 years ago
- Repo for external large-scale work☆6,543Apr 27, 2024Updated last year
- ☆2,950Jan 15, 2026Updated last month
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆241May 12, 2023Updated 2 years ago
- OpenAssistant is a chat-based assistant that understands tasks, can interact with third-party systems, and retrieve information dynamical…☆37,444Aug 17, 2024Updated last year
- JAX-based neural network library☆3,193Mar 2, 2026Updated last week
- Facebook AI Research Sequence-to-Sequence Toolkit written in Python.☆32,176Sep 30, 2025Updated 5 months ago
- A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training☆23,845Aug 15, 2024Updated last year
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆34,987Updated this week
- 🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading☆9,982Sep 7, 2024Updated last year
- Code for the paper "Language Models are Unsupervised Multitask Learners"☆24,668Aug 14, 2024Updated last year
- Running large language models on a single GPU for throughput-oriented scenarios.☆9,381Oct 28, 2024Updated last year
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,738Jan 8, 2024Updated 2 years ago
- RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable)…☆14,393Feb 21, 2026Updated 2 weeks ago
- 🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal model…☆157,462Updated this week
- DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.☆41,759Updated this week
- CodeGen is a family of open-source model for program synthesis. Trained on TPU-v4. Competitive with OpenAI Codex.☆5,170Oct 27, 2025Updated 4 months ago
- Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities☆22,030Jan 23, 2026Updated last month
- Flax is a neural network library for JAX that is designed for flexibility.☆7,105Updated this week
- Code and documentation to train Stanford's Alpaca models, and generate the data.☆30,267Jul 17, 2024Updated last year
- Development repository for the Triton language and compiler☆18,573Updated this week
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆54,071Nov 12, 2025Updated 3 months ago
- StableLM: Stability AI Language Models☆15,756Apr 8, 2024Updated last year
- Ongoing research training transformer models at scale☆15,535Updated this week
- Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"☆6,489Jan 14, 2026Updated last month
- ☆1,636Apr 27, 2023Updated 2 years ago
- Google Research☆37,403Updated this week
- LLM inference in C/C++☆96,322Mar 2, 2026Updated last week
- GPT-3: Language Models are Few-Shot Learners☆15,756Sep 18, 2020Updated 5 years ago
- A framework for training and evaluating AI models on a variety of openly available dialogue datasets.☆10,628Nov 3, 2023Updated 2 years ago
- Open source code for AlphaFold 2.☆14,312Updated this week
- 🤗 The largest hub of ready-to-use datasets for AI models with fast, easy-to-use and efficient data manipulation tools☆21,246Feb 27, 2026Updated last week
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,528Updated this week
- Instruct-tune LLaMA on consumer hardware☆18,964Jul 29, 2024Updated last year
- Pretrained language model with 100B parameters☆3,754Jul 10, 2023Updated 2 years ago
- An open platform for training, serving, and evaluating large language models. Release repo for Vicuna and Chatbot Arena.☆39,426Jun 2, 2025Updated 9 months ago
- A library for efficient similarity search and clustering of dense vectors.☆39,255Updated this week
- A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Auto…☆16,843Updated this week