kingoflolz / mesh-transformer-jax
Model parallel transformers in JAX and Haiku
☆6,298Updated last year
Related projects ⓘ
Alternatives and complementary repositories for mesh-transformer-jax
- An implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed libraries☆6,947Updated this week
- An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.☆8,232Updated 2 years ago
- Repo for external large-scale work☆6,516Updated 6 months ago
- A minimal PyTorch re-implementation of the OpenAI GPT (Generative Pretrained Transformer) training☆20,199Updated 3 months ago
- Instruct-tune LLaMA on consumer hardware☆18,653Updated 3 months ago
- Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM☆7,705Updated 10 months ago
- Running large language models on a single GPU for throughput-oriented scenarios.☆9,198Updated 3 weeks ago
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,502Updated 10 months ago
- ☆2,686Updated this week
- ☆1,508Updated last year
- Large Language Model Text Generation Inference☆9,122Updated this week
- 🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading☆9,248Updated 2 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,409Updated 3 months ago
- Model API for GALACTICA☆2,683Updated last year
- A collection of libraries to optimise AI model performances☆8,375Updated 3 months ago
- OpenLLaMA, a permissively licensed open source reproduction of Meta AI’s LLaMA 7B trained on the RedPajama dataset☆7,384Updated last year
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆5,572Updated 9 months ago
- The simplest way to run LLaMA on your local machine☆13,099Updated 5 months ago
- Home of StarCoder: fine-tuning & inference!☆7,325Updated 8 months ago
- Python package to easily retrain OpenAI's GPT-2 text-generating model on new texts☆3,398Updated last year
- Code and documentation to train Stanford's Alpaca models, and generate the data.☆29,561Updated 4 months ago
- Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Ad…☆5,994Updated 2 months ago
- High-speed download of LLaMA, Facebook's 65B parameter GPT model☆4,169Updated last year
- LLaMA: Open and Efficient Foundation Language Models☆2,807Updated last year
- LLMs build upon Evol Insturct: WizardLM, WizardCoder, WizardMath☆9,269Updated 3 months ago
- CodeGen is a family of open-source model for program synthesis. Trained on TPU-v4. Competitive with OpenAI Codex.☆4,934Updated 8 months ago
- The RedPajama-Data repository contains code for preparing large datasets for training large language models.☆4,571Updated last month
- tiktoken is a fast BPE tokeniser for use with OpenAI's models.☆12,427Updated last month