kingoflolz / mesh-transformer-jaxLinks
Model parallel transformers in JAX and Haiku
☆6,352Updated 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,280Updated 3 weeks ago
- An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.☆8,291Updated 3 years ago
- Repo for external large-scale work☆6,530Updated last year
- 🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading☆9,763Updated 11 months ago
- ☆1,593Updated 2 years ago
- OpenLLaMA, a permissively licensed open source reproduction of Meta AI’s LLaMA 7B trained on the RedPajama dataset☆7,518Updated 2 years ago
- Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch☆11,306Updated last year
- GPT-3: Language Models are Few-Shot Learners☆15,777Updated 4 years ago
- Dataset of GPT-2 outputs for research in detection, biases, and more☆1,995Updated last year
- A collection of libraries to optimise AI model performances☆8,373Updated last year
- Locally run an Instruction-Tuned Chat-Style LLM☆10,213Updated 2 years ago
- High-speed download of LLaMA, Facebook's 65B parameter GPT model☆4,159Updated 2 years ago
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆5,627Updated last year
- CodeGen is a family of open-source model for program synthesis. Trained on TPU-v4. Competitive with OpenAI Codex.☆5,121Updated 6 months ago
- The RedPajama-Data repository contains code for preparing large datasets for training large language models.☆4,791Updated 8 months ago
- Training and serving large-scale neural networks with auto parallelization.☆3,148Updated last year
- ☆2,863Updated 2 months ago
- Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM☆7,853Updated 2 weeks ago
- Code Generation using GPT-J!☆516Updated 3 years 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,897Updated last year
- Running large language models on a single GPU for throughput-oriented scenarios.☆9,356Updated 9 months ago
- Pretrained language model with 100B parameters☆3,755Updated 2 years ago
- A robust Python tool for text-based AI training and generation using GPT-2.☆1,842Updated 2 years ago
- Instruct-tune LLaMA on consumer hardware☆18,948Updated last year
- Code for the paper "Language Models are Unsupervised Multitask Learners"☆24,070Updated last year
- PyTorch package for the discrete VAE used for DALL·E.☆10,878Updated last year
- Trax — Deep Learning with Clear Code and Speed☆8,253Updated 4 months ago
- Model API for GALACTICA☆2,739Updated 2 years ago
- 💥 Fast State-of-the-Art Tokenizers optimized for Research and Production☆9,996Updated 3 weeks ago
- Python package to easily retrain OpenAI's GPT-2 text-generating model on new texts☆3,406Updated 2 years ago