kingoflolz / mesh-transformer-jaxLinks
Model parallel transformers in JAX and Haiku
☆6,357Updated 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,343Updated 2 months ago
- An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.☆8,285Updated 3 years ago
- Repo for external large-scale work☆6,548Updated last year
- ☆1,617Updated 2 years ago
- CodeGen is a family of open-source model for program synthesis. Trained on TPU-v4. Competitive with OpenAI Codex.☆5,157Updated last month
- A robust Python tool for text-based AI training and generation using GPT-2.☆1,843Updated 2 years ago
- Model API for GALACTICA☆2,742Updated 2 years ago
- ☆2,915Updated 2 weeks ago
- Python package to easily retrain OpenAI's GPT-2 text-generating model on new texts☆3,405Updated 2 years ago
- Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM☆7,873Updated last month
- Dataset of GPT-2 outputs for research in detection, biases, and more☆2,006Updated last year
- 🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading☆9,843Updated last year
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆5,634Updated last year
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,729Updated last year
- Training and serving large-scale neural networks with auto parallelization.☆3,165Updated last year
- A collection of libraries to optimise AI model performances☆8,363Updated last year
- 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,888Updated 2 years ago
- ☆2,078Updated 3 years ago
- Large-scale pretraining for dialogue☆2,411Updated 3 years ago
- LLaMA: Open and Efficient Foundation Language Models☆2,800Updated 2 years ago
- Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"☆6,457Updated last month
- Running large language models on a single GPU for throughput-oriented scenarios.☆9,381Updated last year
- Large-scale pretrained models for goal-directed dialog☆887Updated last year
- Code for the paper "Language Models are Unsupervised Multitask Learners"☆1,146Updated 3 years ago
- min(DALL·E) is a fast, minimal port of DALL·E Mini to PyTorch☆3,493Updated 7 months ago
- Instruct-tune LLaMA on consumer hardware☆18,983Updated last year
- 4 bits quantization of LLaMA using GPTQ☆3,081Updated last year
- Beyond the Imitation Game collaborative benchmark for measuring and extrapolating the capabilities of language models