kingoflolz / mesh-transformer-jax
Model parallel transformers in JAX and Haiku
☆6,315Updated last year
Alternatives and similar repositories for mesh-transformer-jax:
Users that are interested in mesh-transformer-jax are comparing it to the libraries listed below
- An implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed libraries☆7,042Updated this week
- Repo for external large-scale work☆6,518Updated 8 months ago
- An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.☆8,260Updated 2 years ago
- Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM☆7,741Updated last week
- ☆2,723Updated this week
- GPT-3: Language Models are Few-Shot Learners☆15,714Updated 4 years ago
- CodeGen is a family of open-source model for program synthesis. Trained on TPU-v4. Competitive with OpenAI Codex.☆4,975Updated 10 months ago
- Running large language models on a single GPU for throughput-oriented scenarios.☆9,254Updated 2 months ago
- 🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading☆9,349Updated 4 months ago
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,567Updated last year
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆5,598Updated 11 months ago
- Locally run an Instruction-Tuned Chat-Style LLM☆10,240Updated last year
- A robust Python tool for text-based AI training and generation using GPT-2.☆1,844Updated last year
- Python package to easily retrain OpenAI's GPT-2 text-generating model on new texts☆3,399Updated 2 years ago
- Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch☆8,155Updated 3 months ago
- Containers for machine learning☆8,284Updated this week
- A collection of libraries to optimise AI model performances☆8,373Updated 5 months ago
- Instruct-tune LLaMA on consumer hardware☆18,758Updated 5 months ago
- ☆1,523Updated last year
- 🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (i…☆8,178Updated this week
- Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Ad…☆6,021Updated 4 months ago
- Training and serving large-scale neural networks with auto parallelization.☆3,093Updated last year
- RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable)…☆13,005Updated last week
- The RedPajama-Data repository contains code for preparing large datasets for training large language models.☆4,616Updated last month
- StableLM: Stability AI Language Models☆15,831Updated 9 months ago
- [ICLR 2024] Fine-tuning LLaMA to follow Instructions within 1 Hour and 1.2M Parameters☆5,799Updated 10 months ago
- Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch☆11,201Updated 8 months ago
- min(DALL·E) is a fast, minimal port of DALL·E Mini to PyTorch☆3,489Updated 2 years ago
- Code for the paper "Language Models are Unsupervised Multitask Learners"☆22,792Updated 5 months ago
- QLoRA: Efficient Finetuning of Quantized LLMs☆10,168Updated 7 months ago