kingoflolz / mesh-transformer-jaxLinks
Model parallel transformers in JAX and Haiku
☆6,333Updated 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,203Updated this week
- An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.☆8,287Updated 3 years ago
- Repo for external large-scale work☆6,527Updated last year
- Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM☆7,832Updated last month
- Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Ad…☆6,060Updated 8 months ago
- ☆1,572Updated 2 years ago
- Running large language models on a single GPU for throughput-oriented scenarios.☆9,320Updated 7 months ago
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,652Updated last year
- ☆2,819Updated this week
- The RedPajama-Data repository contains code for preparing large datasets for training large language models.☆4,727Updated 5 months ago
- A collection of libraries to optimise AI model performances☆8,370Updated 10 months ago
- QLoRA: Efficient Finetuning of Quantized LLMs☆10,462Updated 11 months ago
- 🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading☆9,634Updated 8 months ago
- A robust Python tool for text-based AI training and generation using GPT-2.☆1,842Updated last year
- CodeGen is a family of open-source model for program synthesis. Trained on TPU-v4. Competitive with OpenAI Codex.☆5,087Updated 4 months ago
- An unnecessarily tiny implementation of GPT-2 in NumPy.☆3,360Updated 2 years ago
- OpenLLaMA, a permissively licensed open source reproduction of Meta AI’s LLaMA 7B trained on the RedPajama dataset☆7,498Updated last year
- Large Language Model Text Generation Inference☆10,172Updated this week
- 💥 Fast State-of-the-Art Tokenizers optimized for Research and Production☆9,726Updated this week
- Locally run an Instruction-Tuned Chat-Style LLM☆10,229Updated 2 years ago
- RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable)…☆13,645Updated this week
- Python package to easily retrain OpenAI's GPT-2 text-generating model on new texts☆3,408Updated 2 years ago
- The hub for EleutherAI's work on interpretability and learning dynamics☆2,500Updated last week
- Ongoing research training transformer models at scale☆12,468Updated this week
- Dataset of GPT-2 outputs for research in detection, biases, and more☆1,981Updated last year
- Accessible large language models via k-bit quantization for PyTorch.☆7,088Updated this week
- Home of StarCoder: fine-tuning & inference!☆7,416Updated last year
- StableLM: Stability AI Language Models☆15,832Updated last year
- LLMs build upon Evol Insturct: WizardLM, WizardCoder, WizardMath☆9,416Updated 9 months ago
- Instruct-tune LLaMA on consumer hardware☆18,909Updated 10 months ago