kingoflolz / mesh-transformer-jaxLinks
Model parallel transformers in JAX and Haiku
☆6,342Updated 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,240Updated 2 weeks ago
- An implementation of model parallel GPT-2 and GPT-3-style models using the mesh-tensorflow library.☆8,295Updated 3 years ago
- Implementation of RLHF (Reinforcement Learning with Human Feedback) on top of the PaLM architecture. Basically ChatGPT but with PaLM☆7,851Updated 2 months ago
- CodeGen is a family of open-source model for program synthesis. Trained on TPU-v4. Competitive with OpenAI Codex.☆5,106Updated 5 months ago
- Locally run an Instruction-Tuned Chat-Style LLM☆10,227Updated 2 years ago
- OpenLLaMA, a permissively licensed open source reproduction of Meta AI’s LLaMA 7B trained on the RedPajama dataset☆7,501Updated last year
- Implementation of the LLaMA language model based on nanoGPT. Supports flash attention, Int8 and GPTQ 4bit quantization, LoRA and LLaMA-Ad…☆6,073Updated this week
- 🌸 Run LLMs at home, BitTorrent-style. Fine-tuning and inference up to 10x faster than offloading☆9,694Updated 9 months ago
- Repo for external large-scale work☆6,527Updated last year
- Running large language models on a single GPU for throughput-oriented scenarios.☆9,337Updated 8 months ago
- StableLM: Stability AI Language Models☆15,828Updated last year
- Implementation / replication of DALL-E, OpenAI's Text to Image Transformer, in Pytorch☆5,618Updated last year
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,672Updated last year
- A collection of libraries to optimise AI model performances☆8,372Updated 11 months ago
- The RedPajama-Data repository contains code for preparing large datasets for training large language models.☆4,758Updated 6 months ago
- Instruct-tune LLaMA on consumer hardware☆18,927Updated 11 months ago
- ☆2,839Updated last month
- Training and serving large-scale neural networks with auto parallelization.☆3,137Updated last year
- ☆1,579Updated 2 years ago
- Implementation of DALL-E 2, OpenAI's updated text-to-image synthesis neural network, in Pytorch☆11,297Updated last year
- A list of totally open alternatives to ChatGPT☆4,653Updated 2 years ago
- OpenAssistant is a chat-based assistant that understands tasks, can interact with third-party systems, and retrieve information dynamical…☆37,397Updated 10 months ago
- DALL·E Mini - Generate images from a text prompt☆14,810Updated last year
- LLaMA: Open and Efficient Foundation Language Models☆2,801Updated last year
- QLoRA: Efficient Finetuning of Quantized LLMs☆10,518Updated last year
- Implementation of Imagen, Google's Text-to-Image Neural Network, in Pytorch☆8,311Updated 8 months ago
- 4 bits quantization of LLaMA using GPTQ☆3,053Updated 11 months ago
- Large-scale Self-supervised Pre-training Across Tasks, Languages, and Modalities☆21,461Updated last month
- High-speed download of LLaMA, Facebook's 65B parameter GPT model☆4,161Updated 2 years ago
- Home of CodeT5: Open Code LLMs for Code Understanding and Generation☆3,019Updated last year