bigscience-workshop / Megatron-DeepSpeed
Ongoing research training transformer language models at scale, including: BERT & GPT-2
☆1,378Updated last year
Alternatives and similar repositories for Megatron-DeepSpeed:
Users that are interested in Megatron-DeepSpeed are comparing it to the libraries listed below
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆2,022Updated 3 weeks ago
- Central place for the engineering/scaling WG: documentation, SLURM scripts and logs, compute environment and data.☆988Updated 7 months ago
- Fast Inference Solutions for BLOOM☆562Updated 5 months ago
- [ICML 2024] Break the Sequential Dependency of LLM Inference Using Lookahead Decoding☆1,216Updated 2 weeks ago
- Code for the ICLR 2023 paper "GPTQ: Accurate Post-training Quantization of Generative Pretrained Transformers".☆2,063Updated 11 months ago
- Best practice for training LLaMA models in Megatron-LM☆645Updated last year
- MII makes low-latency and high-throughput inference possible, powered by DeepSpeed.☆1,990Updated this week
- Automatically split your PyTorch models on multiple GPUs for training & inference☆650Updated last year
- distributed trainer for LLMs☆567Updated 10 months ago
- [NIPS2023] RRHF & Wombat☆804Updated last year
- Tutel MoE: An Optimized Mixture-of-Experts Implementation☆786Updated this week
- Open Academic Research on Improving LLaMA to SOTA LLM☆1,617Updated last year
- Crosslingual Generalization through Multitask Finetuning☆529Updated 6 months ago
- ☆1,507Updated last week
- [ICML 2023] SmoothQuant: Accurate and Efficient Post-Training Quantization for Large Language Models☆1,369Updated 8 months ago
- S-LoRA: Serving Thousands of Concurrent LoRA Adapters☆1,801Updated last year
- [ICLR 2024] Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruning☆597Updated last year
- YaRN: Efficient Context Window Extension of Large Language Models☆1,450Updated 11 months ago
- Medusa: Simple Framework for Accelerating LLM Generation with Multiple Decoding Heads☆2,466Updated 8 months ago
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs…☆2,293Updated this week
- Serving multiple LoRA finetuned LLM as one☆1,040Updated 10 months ago
- Scalable toolkit for efficient model alignment☆750Updated this week
- Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackab…☆1,559Updated last year
- Code for our EMNLP 2023 Paper: "LLM-Adapters: An Adapter Family for Parameter-Efficient Fine-Tuning of Large Language Models"☆1,140Updated last year
- Efficient Training (including pre-training and fine-tuning) for Big Models☆580Updated 8 months ago
- Code used for sourcing and cleaning the BigScience ROOTS corpus☆308Updated 2 years ago
- Reading list of Instruction-tuning. A trend starts from Natrural-Instruction (ACL 2022), FLAN (ICLR 2022) and T0 (ICLR 2022).☆766Updated last year
- A simulation framework for RLHF and alternatives. Develop your RLHF method without collecting human data.☆801Updated 8 months ago
- Tools to download and cleanup Common Crawl data☆993Updated last year
- Human preference data for "Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedback"☆1,707Updated last year