Model parallel transformers in JAX and Haiku
☆6,380Jan 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. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- An implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed libraries☆7,457Jun 11, 2026Updated 2 months ago
- An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.☆8,269Feb 25, 2022Updated 4 years ago
- Swarm training framework using Haiku + JAX + Ray for layer parallel transformer language models on unreliable, heterogeneous nodes☆241May 12, 2023Updated 3 years ago
- Repo for external large-scale work☆6,547Apr 27, 2024Updated 2 years ago
- ☆2,997Aug 3, 2026Updated 3 weeks ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- JAX-based neural network library☆3,273Aug 6, 2026Updated 2 weeks ago
- OpenAssistant is a chat-based assistant that understands tasks, can interact with third-party systems, and retrieve information dynamical…☆37,406Aug 17, 2024Updated 2 years ago
- A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training☆24,830Aug 15, 2024Updated 2 years ago
- Facebook AI Research Sequence-to-Sequence Toolkit written in Python.☆32,235Sep 30, 2025Updated 10 months ago
- Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more☆36,209Updated this week
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,754Jan 8, 2024Updated 2 years ago
- Code for the paper "Language Models are Unsupervised Multitask Learners"☆25,018Aug 14, 2024Updated 2 years ago
- 🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading☆10,513Sep 7, 2024Updated last year
- 🤗 Transformers: the model-definition framework for state-of-the-art machine learning models in text, vision, audio, and multimodal model…☆164,394Updated this week
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Running large language models on a single GPU for throughput-oriented scenarios.☆9,352Oct 28, 2024Updated last year
- Flax is a neural network library for JAX that is designed for flexibility.☆7,301Updated this week
- Code and documentation to train Stanford's Alpaca models, and generate the data.☆30,246Jul 17, 2024Updated 2 years ago
- CodeGen is a family of open-source model for program synthesis. Trained on TPU-v4. Competitive with OpenAI Codex.☆5,178Jun 2, 2026Updated 2 months ago
- DeepSpeed is a deep learning optimization library that makes distributed training and inference easy, efficient, and effective.☆42,985Updated this week
- ☆1,672Apr 27, 2023Updated 3 years ago
- Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch☆11,307May 11, 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,680Updated this week
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆5,628Feb 17, 2024Updated 2 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM☆7,867Jul 27, 2026Updated 3 weeks ago
- Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities☆22,194Jan 23, 2026Updated 7 months ago
- Ongoing research training transformer models at scale☆17,563Updated this week
- Development repository for the Triton language and compiler☆19,995Updated this week
- StableLM: Stability AI Language Models☆15,685Apr 8, 2024Updated 2 years ago
- Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"☆6,544Jul 8, 2026Updated last month
- API for the GPT-J language model 🦜. Including a FastAPI backend and a streamlit frontend☆335Oct 25, 2021Updated 4 years ago
- GPT-3: Language Models are Few-Shot Learners☆15,717Sep 18, 2020Updated 5 years ago
- Instruct-tune LLaMA on consumer hardware☆18,907Jul 29, 2024Updated 2 years ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Google Research☆38,611Updated this week
- An open platform for training, serving, and evaluating large language models. Release repo for Vicuna and Chatbot Arena.☆39,524May 1, 2026Updated 3 months ago
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆9,831Updated this week
- A latent text-to-image diffusion model☆73,343Jun 18, 2024Updated 2 years ago
- Training and serving large-scale neural networks with auto parallelization.☆3,178Dec 9, 2023Updated 2 years ago
- Mesh TensorFlow: Model Parallelism Made Easier☆1,630Nov 17, 2023Updated 2 years ago
- A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Auto…☆18,301Updated this week