kingoflolz / mesh-transformer-jaxLinks
Model parallel transformers in JAX and Haiku
☆6,355Updated 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,289Updated last month
- An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.☆8,292Updated 3 years ago
- Repo for external large-scale work☆6,542Updated last year
- Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM☆7,865Updated this week
- ☆1,599Updated 2 years ago
- Code for the paper "Language Models are Unsupervised Multitask Learners"☆24,143Updated last year
- Python package to easily retrain OpenAI's GPT-2 text-generating model on new texts☆3,406Updated 2 years ago
- ☆2,872Updated this week
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,707Updated last year
- Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch☆11,317Updated last year
- CodeGen is a family of open-source model for program synthesis. Trained on TPU-v4. Competitive with OpenAI Codex.☆5,131Updated 7 months ago
- Dataset of GPT-2 outputs for research in detection, biases, and more☆1,994Updated last year
- OpenLLaMA, a permissively licensed open source reproduction of Meta AI’s LLaMA 7B trained on the RedPajama dataset☆7,523Updated 2 years ago
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆5,626Updated last year
- Code for the paper "Exploring the Limits of Transfer Learning with a Unified Text-to-Text Transformer"☆6,418Updated 4 months ago
- ☆2,068Updated 3 years ago
- A robust Python tool for text-based AI training and generation using GPT-2.☆1,840Updated 2 years ago
- Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch☆8,359Updated 11 months ago
- 🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading☆9,787Updated last year
- StableLM: Stability AI Language Models☆15,804Updated last year
- A collection of libraries to optimise AI model performances☆8,369Updated last year
- Beyond the Imitation Game collaborative benchmark for measuring and extrapolating the capabilities of language models☆3,111Updated last year
- Drive a browser with GPT-3☆1,927Updated 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,894Updated last year
- Code for the paper "Language Models are Unsupervised Multitask Learners"☆1,148Updated 2 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
- RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable)…☆13,950Updated last week
- ☆4,558Updated 2 years ago
- QLoRA: Efficient Finetuning of Quantized LLMs☆10,648Updated last year
- Training and serving large-scale neural networks with auto parallelization.☆3,153Updated last year