kingoflolz / mesh-transformer-jaxLinks
Model parallel transformers in JAX and Haiku
☆6,360Updated 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,367Updated last month
- 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,547Updated 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,887Updated 2 years ago
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆5,633Updated last year
- Python package to easily retrain OpenAI's GPT-2 text-generating model on new texts☆3,403Updated 3 years ago
- ☆1,627Updated 2 years ago
- A robust Python tool for text-based AI training and generation using GPT-2.☆1,842Updated 2 years ago
- ☆2,933Updated last week
- CodeGen is a family of open-source model for program synthesis. Trained on TPU-v4. Competitive with OpenAI Codex.☆5,166Updated 2 months ago
- Dataset of GPT-2 outputs for research in detection, biases, and more☆2,013Updated 2 years ago
- GPT-3: Language Models are Few-Shot Learners☆15,765Updated 5 years ago
- Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch☆11,340Updated last year
- Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Ad…☆6,092Updated 6 months ago
- Code for the paper "Language Models are Unsupervised Multitask Learners"☆24,546Updated last year
- min(DALL·E) is a fast, minimal port of DALL·E Mini to PyTorch☆3,496Updated 8 months ago
- Code and documentation to train Stanford's Alpaca models, and generate the data.☆30,269Updated last year
- PyTorch package for the discrete VAE used for DALL·E.☆10,873Updated last year
- High-speed download of LLaMA, Facebook's 65B parameter GPT model☆4,145Updated 2 years ago
- A collection of libraries to optimise AI model performances☆8,355Updated last year
- Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM☆7,877Updated 3 months ago
- Code for the paper "Language Models are Unsupervised Multitask Learners"☆1,145Updated 3 years ago
- Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion☆7,759Updated 3 years ago
- Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"☆6,476Updated this week
- Simple command line tool for text to image generation using OpenAI's CLIP and Siren (Implicit neural representation network). Technique w…☆4,334Updated 3 years ago
- Large-scale pretraining for dialogue☆2,418Updated 3 years ago
- Official implementation for "Multimodal Chain-of-Thought Reasoning in Language Models" (stay tuned and more will be updated)☆3,987Updated last year
- 🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading☆9,871Updated last year
- Model API for GALACTICA☆2,737Updated 2 years ago
- Open clone of OpenAI's unreleased WebText dataset scraper. This version uses pushshift.io files instead of the API for speed.☆750Updated 3 years ago