bigscience-workshop / Megatron-DeepSpeed
Ongoing research training transformer language models at scale, including: BERT & GPT-2
☆1,354Updated 9 months ago
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☆1,957Updated 3 weeks ago
- Fast Inference Solutions for BLOOM☆563Updated 3 months ago
- Central place for the engineering/scaling WG: documentation, SLURM scripts and logs, compute environment and data.☆982Updated 5 months ago
- Open Academic Research on Improving LLaMA to SOTA LLM☆1,615Updated last year
- [ICML 2024] Break the Sequential Dependency of LLM Inference Using Lookahead Decoding☆1,179Updated 3 months ago
- Code for the ICLR 2023 paper "GPTQ: Accurate Post-training Quantization of Generative Pretrained Transformers".☆1,998Updated 9 months ago
- Best practice for training LLaMA models in Megatron-LM☆638Updated last year
- [NIPS2023] RRHF & Wombat☆802Updated last year
- distributed trainer for LLMs☆555Updated 7 months ago
- LOMO: LOw-Memory Optimization☆979Updated 6 months ago
- Tutel MoE: An Optimized Mixture-of-Experts Implementation☆746Updated this week
- Code for our EMNLP 2023 Paper: "LLM-Adapters: An Adapter Family for Parameter-Efficient Fine-Tuning of Large Language Models"☆1,108Updated 10 months ago
- MII makes low-latency and high-throughput inference possible, powered by DeepSpeed.☆1,942Updated last month
- Automatically split your PyTorch models on multiple GPUs for training & inference☆643Updated last year
- Efficient Training (including pre-training and fine-tuning) for Big Models☆574Updated 5 months ago
- S-LoRA: Serving Thousands of Concurrent LoRA Adapters☆1,777Updated 11 months ago
- DeepSeekMoE: Towards Ultimate Expert Specialization in Mixture-of-Experts Language Models☆1,083Updated last year
- A library for accelerating Transformer models on NVIDIA GPUs, including using 8-bit floating point (FP8) precision on Hopper and Ada GPUs…☆2,086Updated this week
- Crosslingual Generalization through Multitask Finetuning☆521Updated 3 months ago
- ⛷️ LLaMA-MoE: Building Mixture-of-Experts from LLaMA with Continual Pre-training (EMNLP 2024)☆908Updated last month
- ☆1,486Updated 2 months ago
- YaRN: Efficient Context Window Extension of Large Language Models☆1,398Updated 9 months ago
- [ICLR 2024] Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruning☆578Updated 10 months ago
- Scalable toolkit for efficient model alignment☆674Updated this week
- Human preference data for "Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedback"☆1,667Updated last year
- A plug-and-play library for parameter-efficient-tuning (Delta Tuning)☆1,009Updated 3 months ago
- Secrets of RLHF in Large Language Models Part I: PPO☆1,320Updated 10 months ago
- A modular RL library to fine-tune language models to human preferences☆2,253Updated 10 months ago
- Reference implementation for DPO (Direct Preference Optimization)☆2,323Updated 5 months ago