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,323Updated last month
- An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.☆8,287Updated 3 years ago
- Repo for external large-scale work☆6,546Updated last year
- CodeGen is a family of open-source model for program synthesis. Trained on TPU-v4. Competitive with OpenAI Codex.☆5,143Updated 8 months ago
- Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM☆7,869Updated 2 weeks ago
- Python package to easily retrain OpenAI's GPT-2 text-generating model on new texts☆3,406Updated 2 years ago
- ☆1,609Updated 2 years ago
- A robust Python tool for text-based AI training and generation using GPT-2.☆1,841Updated 2 years ago
- Training and serving large-scale neural networks with auto parallelization.☆3,159Updated last year
- 🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading☆9,820Updated last year
- A collection of libraries to optimise AI model performances☆8,367Updated last year
- ☆2,896Updated this week
- StableLM: Stability AI Language Models☆15,797Updated 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
- GPT-3: Language Models are Few-Shot Learners☆15,781Updated 5 years ago
- ☆9,014Updated last year
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,716Updated last year
- The simplest way to run LLaMA on your local machine☆13,043Updated last year
- Code for the paper "Language Models are Unsupervised Multitask Learners"☆24,320Updated last year
- Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Ad…☆6,079Updated 3 months ago
- OpenAssistant is a chat-based assistant that understands tasks, can interact with third-party systems, and retrieve information dynamical…☆37,480Updated last year
- The RedPajama-Data repository contains code for preparing large datasets for training large language models.☆4,834Updated 10 months ago
- High-speed download of LLaMA, Facebook's 65B parameter GPT model☆4,154Updated 2 years ago
- Instruct-tune LLaMA on consumer hardware☆18,977Updated last year
- Build, personalize and control your own LLMs. From data pre-processing to fine-tuning, xTuring provides an easy way to personalize open-s…☆2,658Updated this week
- Locally run an Instruction-Tuned Chat-Style LLM☆10,204Updated 2 years ago
- Drive a browser with GPT-3☆1,932Updated last year
- ☆2,174Updated 2 years ago
- Containers for machine learning☆8,880Updated last week
- Running large language models on a single GPU for throughput-oriented scenarios.☆9,371Updated last year