ROCm / Megatron-LMLinks
Ongoing research training transformer models at scale
☆29Updated this week
Alternatives and similar repositories for Megatron-LM
Users that are interested in Megatron-LM are comparing it to the libraries listed below
Sorting:
- Microsoft Collective Communication Library☆66Updated 10 months ago
- Intel® Extension for DeepSpeed* is an extension to DeepSpeed that brings feature support with SYCL kernels on Intel GPU(XPU) device. Note…☆63Updated 3 months ago
- MAD (Model Automation and Dashboarding)☆25Updated last week
- ☆48Updated this week
- ☆92Updated 11 months ago
- A high-throughput and memory-efficient inference and serving engine for LLMs☆83Updated this week
- A high-throughput and memory-efficient inference and serving engine for LLMs☆108Updated this week
- QuickReduce is a performant all-reduce library designed for AMD ROCm that supports inline compression.☆33Updated last month
- nnScaler: Compiling DNN models for Parallel Training☆118Updated 3 weeks ago
- ☆124Updated 9 months ago
- An efficient GPU support for LLM inference with x-bit quantization (e.g. FP6,FP5).☆265Updated 3 months ago
- RCCL Performance Benchmark Tests☆75Updated last week
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆83Updated last week
- ☆72Updated last year
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆221Updated 2 years ago
- ☆46Updated 10 months ago
- PyTorch bindings for CUTLASS grouped GEMM.☆124Updated 4 months ago
- Development repository for the Triton language and compiler☆135Updated this week
- AI Tensor Engine for ROCm☆285Updated last week
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆216Updated last year
- Fast and memory-efficient exact attention☆193Updated last week
- Chimera: bidirectional pipeline parallelism for efficiently training large-scale models.☆66Updated 7 months ago
- A lightweight design for computation-communication overlap.☆181Updated last week
- oneCCL Bindings for Pytorch*☆102Updated 2 months ago
- ☆25Updated this week
- Thunder Research Group's Collective Communication Library☆42Updated 3 months ago
- Applied AI experiments and examples for PyTorch☆299Updated last month
- (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.☆41Updated 2 years ago
- FlexFlow Serve: Low-Latency, High-Performance LLM Serving☆63Updated last month
- ☆75Updated 4 years ago