kingoflolz / mesh-transformer-jax
Model parallel transformers in JAX and Haiku
☆6,321Updated 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,134Updated last week
- 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,519Updated 10 months ago
- OpenLLaMA, a permissively licensed open source reproduction of Meta AI’s LLaMA 7B trained on the RedPajama dataset☆7,462Updated last year
- Running large language models on a single GPU for throughput-oriented scenarios.☆9,278Updated 4 months ago
- Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch☆11,237Updated 10 months ago
- ☆2,767Updated 2 weeks ago
- Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Ad…☆6,043Updated 6 months ago
- Large language models (LLMs) made easy, EasyLM is a one stop solution for pre-training, finetuning, evaluating and serving LLMs in JAX/Fl…☆2,463Updated 7 months ago
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,603Updated last year
- ☆1,547Updated last year
- Instruct-tune LLaMA on consumer hardware☆18,842Updated 7 months ago
- Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM☆7,765Updated last month
- Accessible large language models via k-bit quantization for PyTorch.☆6,818Updated this week
- A collection of libraries to optimise AI model performances☆8,373Updated 8 months ago
- The simplest way to run LLaMA on your local machine☆13,083Updated 9 months ago
- Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) with Stable Diffusion☆7,670Updated 2 years ago
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆5,605Updated last year
- 🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading☆9,508Updated 6 months ago
- Dataset of GPT-2 outputs for research in detection, biases, and more☆1,967Updated last year
- Implementation of Dreambooth (https://arxiv.org/abs/2208.12242) by way of Textual Inversion (https://arxiv.org/abs/2208.01618) for Stable…☆3,209Updated last year
- High-speed download of LLaMA, Facebook's 65B parameter GPT model☆4,164Updated last year
- QLoRA: Efficient Finetuning of Quantized LLMs☆10,322Updated 9 months ago
- Training and serving large-scale neural networks with auto parallelization.☆3,115Updated last year
- RWKV (pronounced RwaKuv) is an RNN with great LLM performance, which can also be directly trained like a GPT transformer (parallelizable)…☆13,374Updated this week
- StableLM: Stability AI Language Models☆15,836Updated 11 months ago
- AITemplate is a Python framework which renders neural network into high performance CUDA/HIP C++ code. Specialized for FP16 TensorCore (N…☆4,620Updated 3 months ago
- Official implementation for "Multimodal Chain-of-Thought Reasoning in Language Models" (stay tuned and more will be updated)☆3,889Updated 9 months ago
- The RedPajama-Data repository contains code for preparing large datasets for training large language models.☆4,678Updated 3 months ago
- 4 bits quantization of LLaMA using GPTQ☆3,045Updated 8 months ago