kingoflolz / mesh-transformer-jaxLinks
Model parallel transformers in JAX and Haiku
☆6,351Updated 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,303Updated this week
- An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.☆8,293Updated 3 years ago
- Repo for external large-scale work☆6,547Updated last year
- ☆2,882Updated 2 weeks ago
- ☆1,600Updated 2 years ago
- CodeGen is a family of open-source model for program synthesis. Trained on TPU-v4. Competitive with OpenAI Codex.☆5,133Updated 7 months ago
- Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM☆7,864Updated 2 weeks ago
- Python package to easily retrain OpenAI's GPT-2 text-generating model on new texts☆3,407Updated 2 years ago
- Running large language models on a single GPU for throughput-oriented scenarios.☆9,364Updated 10 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,893Updated last year
- Guide to using pre-trained large language models of source code☆1,838Updated last year
- A robust Python tool for text-based AI training and generation using GPT-2.☆1,841Updated 2 years ago
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆5,628Updated last year
- GPT-3: Language Models are Few-Shot Learners☆15,781Updated 4 years ago
- Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Ad…☆6,079Updated 2 months ago
- 🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading☆9,791Updated last year
- Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"☆6,421Updated 4 months ago
- Code for the paper "Jukebox: A Generative Model for Music"☆8,015Updated last year
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,710Updated last year
- Public release of the TransCoder research project https://arxiv.org/pdf/2006.03511.pdf☆1,722Updated 3 years ago
- OpenLLaMA, a permissively licensed open source reproduction of Meta AI’s LLaMA 7B trained on the RedPajama dataset☆7,529Updated 2 years ago
- LLaMA: Open and Efficient Foundation Language Models☆2,803Updated last year
- The RedPajama-Data repository contains code for preparing large datasets for training large language models.☆4,807Updated 9 months ago
- Code for GPT-4chan☆637Updated 3 years ago
- Code Generation using GPT-J!☆515Updated 3 years ago
- Training and serving large-scale neural networks with auto parallelization.☆3,153Updated last year
- Large-scale pretraining for dialogue☆2,404Updated 2 years ago
- Dataset of GPT-2 outputs for research in detection, biases, and more☆1,994Updated last year
- 💥 Fast State-of-the-Art Tokenizers optimized for Research and Production☆10,060Updated 2 weeks ago
- High-speed download of LLaMA, Facebook's 65B parameter GPT model☆4,158Updated 2 years ago