kingoflolz / mesh-transformer-jaxLinks
Model parallel transformers in JAX and Haiku
☆6,338Updated 2 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,228Updated last week
- An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.☆8,295Updated 3 years ago
- Repo for external large-scale work☆6,530Updated last year
- A collection of libraries to optimise AI model performances☆8,372Updated 11 months ago
- 🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading☆9,675Updated 9 months ago
- OpenLLaMA, a permissively licensed open source reproduction of Meta AI’s LLaMA 7B trained on the RedPajama dataset☆7,498Updated last year
- ☆2,833Updated 2 weeks ago
- Running large language models on a single GPU for throughput-oriented scenarios.☆9,338Updated 7 months ago
- Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM☆7,847Updated last month
- A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training☆22,108Updated 10 months ago
- LLaMA: Open and Efficient Foundation Language Models☆2,802Updated last year
- Instruct-tune LLaMA on consumer hardware☆18,917Updated 10 months ago
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,670Updated last year
- A robust Python tool for text-based AI training and generation using GPT-2.☆1,842Updated last year
- Locally run an Instruction-Tuned Chat-Style LLM☆10,227Updated 2 years ago
- Python package to easily retrain OpenAI's GPT-2 text-generating model on new texts☆3,406Updated 2 years ago
- min(DALL·E) is a fast, minimal port of DALL·E Mini to PyTorch☆3,490Updated last month
- Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Ad…☆6,073Updated 9 months ago
- Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"☆6,379Updated last month
- Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch☆11,291Updated last year
- An unnecessarily tiny implementation of GPT-2 in NumPy.☆3,368Updated 2 years ago
- High-speed download of LLaMA, Facebook's 65B parameter GPT model☆4,163Updated last year
- StableLM: Stability AI Language Models☆15,828Updated last year
- Point cloud diffusion for 3D model synthesis☆6,729Updated 11 months ago
- Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Fl…☆2,479Updated 10 months ago
- [ICLR 2024] Fine-tuning LLaMA to follow Instructions within 1 Hour and 1.2M Parameters☆5,882Updated last year
- Training and serving large-scale neural networks with auto parallelization.☆3,138Updated last year
- Accessible large language models via k-bit quantization for PyTorch.☆7,142Updated this week
- The simplest way to run LLaMA on your local machine☆13,070Updated last year
- QLoRA: Efficient Finetuning of Quantized LLMs☆10,500Updated last year