Automated Parallelization System and Infrastructure for Multiple Ecosystems
☆82Nov 19, 2024Updated last year
Alternatives and similar repositories for easydist
Users that are interested in easydist are comparing it to the libraries listed below
Sorting:
- TePDist (TEnsor Program DISTributed) is an HLO-level automatic distributed system for DL models.☆99Apr 22, 2023Updated 2 years ago
- Accelerate Video Diffusion Inference via Sketching-Rendering Cooperation☆19Jun 11, 2025Updated 8 months ago
- The official implementation of "Helen: Optimizing CTR Prediction Models with Frequency-wise Hessian Eigenvalue Regularization"☆16Mar 14, 2024Updated last year
- Chimera: bidirectional pipeline parallelism for efficiently training large-scale models.☆70Mar 20, 2025Updated 11 months ago
- ☆23Aug 21, 2025Updated 6 months ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆114Sep 10, 2024Updated last year
- 面向多平台编译优化的深度学习中间表示☆10Oct 28, 2024Updated last year
- A baseline repository of Auto-Parallelism in Training Neural Networks☆147Jun 25, 2022Updated 3 years ago
- A fast communication-overlapping library for tensor/expert parallelism on GPUs.☆1,261Aug 28, 2025Updated 6 months ago
- ☆24Feb 20, 2024Updated 2 years ago
- Pipeline Parallelism for PyTorch☆785Aug 21, 2024Updated last year
- Distributed SDDMM Kernel☆12Jul 8, 2022Updated 3 years ago
- TACOS: [T]opology-[A]ware [Co]llective Algorithm [S]ynthesizer for Distributed Machine Learning☆32Jun 13, 2025Updated 8 months ago
- Democratizing AlphaFold3: an PyTorch reimplementation to accelerate protein structure prediction☆55Dec 16, 2024Updated last year
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆475Updated this week
- BladeDISC is an end-to-end DynamIc Shape Compiler project for machine learning workloads.☆918Dec 30, 2024Updated last year
- Zero Bubble Pipeline Parallelism☆451May 7, 2025Updated 9 months ago
- A library to analyze PyTorch traces.☆467Feb 4, 2026Updated last month
- Distributed Compiler based on Triton for Parallel Systems☆1,371Feb 13, 2026Updated 2 weeks ago
- Example of applying CUDA graphs to LLaMA-v2☆12Aug 25, 2023Updated 2 years ago
- A high-throughput and memory-efficient inference and serving engine for LLMs☆13Feb 11, 2026Updated 2 weeks ago
- [MLSys 2023] Pre-train and Search: Efficient Embedding Table Sharding with Pre-trained Neural Cost Models☆16May 5, 2023Updated 2 years ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆644Jan 15, 2026Updated last month
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆106Jun 28, 2025Updated 8 months ago
- Examples of CUDA implementations by Cutlass CuTe☆269Jul 1, 2025Updated 8 months ago
- Model-Based Uncertainty in Value Functions (AISTATS2023)☆16Feb 28, 2023Updated 3 years ago
- ☆14Aug 29, 2023Updated 2 years ago
- ☆65Apr 26, 2025Updated 10 months ago
- (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.☆44Nov 4, 2022Updated 3 years ago
- Keyformer proposes KV Cache reduction through key tokens identification and without the need for fine-tuning☆57Mar 26, 2024Updated last year
- Artifact of OSDI '24 paper, ”Llumnix: Dynamic Scheduling for Large Language Model Serving“☆64Jun 5, 2024Updated last year
- Scalable PaLM implementation of PyTorch☆190Dec 19, 2022Updated 3 years ago
- This repository organizes the Imagnet1k dataset into 10 coarse classes, where each class consists of semantically similar image categorie…☆22Dec 11, 2023Updated 2 years ago
- NCCL Profiling Kit☆152Jul 1, 2024Updated last year
- Large-scale model inference.☆627Sep 12, 2023Updated 2 years ago
- CS294 AI Systems Class Website☆17Apr 25, 2022Updated 3 years ago
- ☆41Oct 15, 2025Updated 4 months ago
- PaperHelper: Knowledge-Based LLM QA Paper Reading Assistant with Reliable References☆20Jun 13, 2024Updated last year
- Minimal Decision Transformer Implementation written in Jax (Flax).☆17Aug 8, 2022Updated 3 years ago