kingoflolz / mesh-transformer-jax
Model parallel transformers in JAX and Haiku
☆6,329Updated 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
- An implementation of model parallel autoregressive transformers on GPUs, based on the Megatron and DeepSpeed libraries☆7,172Updated this week
- CodeGen is a family of open-source model for program synthesis. Trained on TPU-v4. Competitive with OpenAI Codex.☆5,075Updated 3 months ago
- An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.☆8,286Updated 3 years ago
- Repo for external large-scale work☆6,524Updated last year
- Running large language models on a single GPU for throughput-oriented scenarios.☆9,311Updated 6 months ago
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,633Updated last year
- 🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading☆9,605Updated 8 months ago
- ☆1,563Updated 2 years ago
- ☆2,800Updated this week
- Instruct-tune LLaMA on consumer hardware☆18,902Updated 9 months ago
- Training and serving large-scale neural networks with auto parallelization.☆3,129Updated last year
- A collection of libraries to optimise AI model performances☆8,370Updated 9 months ago
- OpenLLaMA, a permissively licensed open source reproduction of Meta AI’s LLaMA 7B trained on the RedPajama dataset☆7,482Updated last year
- Accessible large language models via k-bit quantization for PyTorch.☆6,972Updated this week
- The RedPajama-Data repository contains code for preparing large datasets for training large language models.☆4,711Updated 4 months ago
- Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Ad…☆6,052Updated 8 months ago
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,630Updated last month
- Development repository for the Triton language and compiler☆15,447Updated this week
- StableLM: Stability AI Language Models☆15,832Updated last year
- Ongoing research training transformer models at scale☆12,261Updated this week
- QLoRA: Efficient Finetuning of Quantized LLMs☆10,417Updated 10 months ago
- Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM☆7,794Updated last week
- min(DALL·E) is a fast, minimal port of DALL·E Mini to PyTorch☆3,489Updated last week
- 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,892Updated last year
- Tensor library for machine learning☆12,445Updated this week
- Locally run an Instruction-Tuned Chat-Style LLM☆10,233Updated 2 years ago
- General technology for enabling AI capabilities w/ LLMs and MLLMs☆3,968Updated 2 weeks ago
- Code for the paper "Evaluating Large Language Models Trained on Code"☆2,719Updated 3 months ago
- High-speed download of LLaMA, Facebook's 65B parameter GPT model☆4,165Updated last year
- Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities☆21,171Updated 2 months ago