kingoflolz / mesh-transformer-jaxLinks
Model parallel transformers in JAX and Haiku
☆6,355Updated 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,350Updated last week
- An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.☆8,286Updated 3 years ago
- Repo for external large-scale work☆6,547Updated last year
- Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM☆7,870Updated 2 months ago
- A collection of libraries to optimise AI model performances☆8,363Updated last year
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,732Updated last year
- ☆2,920Updated last week
- ☆4,555Updated 2 years ago
- 🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading☆9,858Updated last year
- StableLM: Stability AI Language Models☆15,787Updated last year
- Locally run an Instruction-Tuned Chat-Style LLM☆10,194Updated 2 years ago
- Running large language models on a single GPU for throughput-oriented scenarios.☆9,383Updated last year
- OpenLLaMA, a permissively licensed open source reproduction of Meta AI’s LLaMA 7B trained on the RedPajama dataset☆7,527Updated 2 years ago
- ☆1,622Updated 2 years ago
- Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Ad…☆6,091Updated 5 months ago
- The goal of this project is to enable users to create cool web demos using the newly released OpenAI GPT-3 API with just a few lines of P…☆2,888Updated 2 years ago
- High-speed download of LLaMA, Facebook's 65B parameter GPT model☆4,152Updated 2 years ago
- min(DALL·E) is a fast, minimal port of DALL·E Mini to PyTorch☆3,492Updated 7 months ago
- RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable)…☆14,233Updated this week
- An unnecessarily tiny implementation of GPT-2 in NumPy.☆3,431Updated 2 years ago
- The RedPajama-Data repository contains code for preparing large datasets for training large language models.☆4,905Updated last year
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆5,631Updated last year
- QLoRA: Efficient Finetuning of Quantized LLMs☆10,786Updated last year
- The simplest way to run LLaMA on your local machine☆13,019Updated last year
- Build, personalize and control your own LLMs. From data pre-processing to fine-tuning, xTuring provides an easy way to personalize open-s…☆2,664Updated 3 weeks ago
- LLaMA: Open and Efficient Foundation Language Models☆2,800Updated 2 years ago
- Training and serving large-scale neural networks with auto parallelization.☆3,171Updated 2 years ago
- Drive a browser with GPT-3☆1,933Updated last year
- Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Fl…☆2,503Updated last year
- CodeGen is a family of open-source model for program synthesis. Trained on TPU-v4. Competitive with OpenAI Codex.☆5,157Updated last month