kingoflolz / mesh-transformer-jax
Model parallel transformers in JAX and Haiku
☆6,334Updated 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,178Updated last week
- An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.☆8,289Updated 3 years ago
- Repo for external large-scale work☆6,527Updated last year
- 🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading☆9,613Updated 8 months ago
- ☆2,807Updated 2 weeks ago
- OpenLLaMA, a permissively licensed open source reproduction of Meta AI’s LLaMA 7B trained on the RedPajama dataset☆7,484Updated last year
- Running large language models on a single GPU for throughput-oriented scenarios.☆9,316Updated 6 months ago
- ☆1,565Updated 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,054Updated 8 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,476Updated 9 months ago
- CodeGen is a family of open-source model for program synthesis. Trained on TPU-v4. Competitive with OpenAI Codex.☆5,082Updated 3 months ago
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,641Updated last year
- QLoRA: Efficient Finetuning of Quantized LLMs☆10,423Updated 11 months ago
- ☆2,135Updated last year
- Model API for GALACTICA☆2,723Updated 2 years ago
- A scalable generative AI framework built for researchers and developers working on Large Language Models, Multimodal, and Speech AI (Auto…☆14,317Updated this week
- Instruct-tune LLaMA on consumer hardware☆18,900Updated 9 months ago
- PyTorch package for the discrete VAE used for DALL·E.☆10,854Updated last year
- The RedPajama-Data repository contains code for preparing large datasets for training large language models.☆4,713Updated 5 months ago
- Training and serving large-scale neural networks with auto parallelization.☆3,131Updated last year
- 4 bits quantization of LLaMA using GPTQ☆3,050Updated 10 months ago
- 💥 Fast State-of-the-Art Tokenizers optimized for Research and Production☆9,670Updated last month
- Mesh TensorFlow: Model Parallelism Made Easier☆1,606Updated last year
- RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable)…☆13,590Updated this week
- Locally run an Instruction-Tuned Chat-Style LLM☆10,231Updated 2 years ago
- StableLM: Stability AI Language Models☆15,832Updated last year
- A collection of libraries to optimise AI model performances☆8,371Updated 9 months ago
- tiktoken is a fast BPE tokeniser for use with OpenAI's models.☆14,461Updated last month
- Code and documentation to train Stanford's Alpaca models, and generate the data.☆29,990Updated 9 months ago
- LLaMA: Open and Efficient Foundation Language Models☆2,800Updated last year