bigscience-workshop / Megatron-DeepSpeed
Ongoing research training transformer language models at scale, including: BERT & GPT-2
☆1,386Updated 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,049Updated last month
- Central place for the engineering/scaling WG: documentation, SLURM scripts and logs, compute environment and data.☆992Updated 8 months ago
- Fast Inference Solutions for BLOOM☆561Updated 6 months ago
- Best practice for training LLaMA models in Megatron-LM☆649Updated last year
- MII makes low-latency and high-throughput inference possible, powered by DeepSpeed.☆2,002Updated 3 weeks ago
- Efficient Training (including pre-training and fine-tuning) for Big Models☆584Updated this week
- Crosslingual Generalization through Multitask Finetuning☆531Updated 7 months ago
- Code for the ICLR 2023 paper "GPTQ: Accurate Post-training Quantization of Generative Pretrained Transformers".☆2,093Updated last year
- Tutel MoE: Optimized Mixture-of-Experts Library, Support DeepSeek FP8/FP4☆803Updated this week
- Automatically split your PyTorch models on multiple GPUs for training & inference☆652Updated last year
- distributed trainer for LLMs☆572Updated 11 months ago
- [ICML 2024] Break the Sequential Dependency of LLM Inference Using Lookahead Decoding☆1,242Updated last month
- Scalable toolkit for efficient model alignment☆770Updated last week
- Open Academic Research on Improving LLaMA to SOTA LLM☆1,618Updated last year
- [NIPS2023] RRHF & Wombat☆806Updated last year
- ☆1,512Updated last week
- Code used for sourcing and cleaning the BigScience ROOTS corpus☆309Updated 2 years ago
- [ICLR 2024] Sheared LLaMA: Accelerating Language Model Pre-training via Structured Pruning☆598Updated last year
- Human preference data for "Training a Helpful and Harmless Assistant with Reinforcement Learning from Human Feedback"☆1,723Updated last year
- Code for the paper Fine-Tuning Language Models from Human Preferences☆1,320Updated last year
- Secrets of RLHF in Large Language Models Part I: PPO☆1,357Updated last year
- Minimalistic large language model 3D-parallelism training☆1,793Updated this week
- YaRN: Efficient Context Window Extension of Large Language Models☆1,470Updated last year
- Expanding natural instructions☆995Updated last year
- The official repo of Pai-Megatron-Patch for LLM & VLM large scale training developed by Alibaba Cloud.☆1,007Updated this week
- train llama on a single A100 80G node using 🤗 transformers and 🚀 Deepspeed Pipeline Parallelism☆218Updated last year
- ☆749Updated 10 months ago
- 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,565Updated last year
- ⛷️ LLaMA-MoE: Building Mixture-of-Experts from LLaMA with Continual Pre-training (EMNLP 2024)☆956Updated 4 months ago
- A modular RL library to fine-tune language models to human preferences☆2,301Updated last year