Inference framework for MoE layers based on TensorRT with Python binding
☆41May 31, 2021Updated 4 years ago
Alternatives and similar repositories for InfMoE
Users that are interested in InfMoE are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Compiler for Dynamic Neural Networks☆45Nov 13, 2023Updated 2 years ago
- ☆89Apr 2, 2022Updated 3 years ago
- ☆38Jun 27, 2025Updated 8 months ago
- An agent for CUDA compute-communication kernel co-design☆32Mar 11, 2026Updated last week
- Official resporitory for "IPDPS' 24 QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devices".☆20Feb 23, 2024Updated 2 years ago
- PilotFish harvests the free GPU cycles of cloud gaming with deep learning training☆14Jul 2, 2022Updated 3 years ago
- Adaptive Message Quantization and Parallelization for Distributed Full-graph GNN Training☆24Mar 1, 2024Updated 2 years ago
- A fast MoE impl for PyTorch☆1,846Feb 10, 2025Updated last year
- Explore Inter-layer Expert Affinity in MoE Model Inference☆16May 6, 2024Updated last year
- A language and compiler for irregular tensor programs.☆152Nov 29, 2024Updated last year
- Surrogate-based Hyperparameter Tuning System☆29Jun 29, 2023Updated 2 years ago
- A simple MIPS CPU for BUAA CO course (and now NSCSCC).☆10May 15, 2021Updated 4 years ago
- A hybrid partitioner based quantum circuit simulation system on GPU☆48Aug 17, 2022Updated 3 years ago
- ☆11Apr 3, 2023Updated 2 years ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆125Jun 23, 2022Updated 3 years ago
- A plug-in of Microsoft DeepSpeed to fix the bug of DeepSpeed pipeline☆25Apr 16, 2021Updated 4 years ago
- Binary translation in Rust☆13Jun 22, 2020Updated 5 years ago
- A router IP written in Verilog.☆12Dec 20, 2019Updated 6 years ago
- A hand-written recursive decent Verilog parser.☆10Jan 30, 2026Updated last month
- Compiling finite generators to digital logic. WIP☆13Aug 24, 2020Updated 5 years ago
- Code for CPM-2 Pre-Train☆157Mar 18, 2023Updated 3 years ago
- PKU CompNet'19 Lab 2 - Homebrew TCP☆12Nov 29, 2019Updated 6 years ago
- Write pandoc markdown in OverLeaf☆12Sep 28, 2022Updated 3 years ago
- ☆85Feb 5, 2026Updated last month
- An Attention Superoptimizer☆22Jan 20, 2025Updated last year
- [ICDCS 2023] Evaluation and Optimization of Gradient Compression for Distributed Deep Learning☆10Apr 28, 2023Updated 2 years ago
- Official Repo for "SplitQuant / LLM-PQ: Resource-Efficient LLM Offline Serving on Heterogeneous GPUs via Phase-Aware Model Partition and …☆37Aug 29, 2025Updated 6 months ago
- Benchmark PyTorch Custom Operators☆14Jul 6, 2023Updated 2 years ago
- Documentation for TCP Lab☆12May 20, 2025Updated 10 months ago
- A Cluster-Wide Model Manager to Accelerate DNN Training via Automated Training Warmup☆36Jan 9, 2023Updated 3 years ago
- Primo: Practical Learning-Augmented Systems with Interpretable Models☆19Dec 26, 2023Updated 2 years ago
- Artifact for "Apparate: Rethinking Early Exits to Tame Latency-Throughput Tensions in ML Serving" [SOSP '24]☆24Nov 21, 2024Updated last year
- ☆21May 13, 2022Updated 3 years ago
- This package implements THOR: Transformer with Stochastic Experts.☆64Oct 7, 2021Updated 4 years ago
- A lightweight, user-friendly data-plane for LLM training.☆38Sep 10, 2025Updated 6 months ago
- Let's discover a new world. — Edit☆10Jan 6, 2017Updated 9 years ago
- This is the implementation repository of our SOSP'24 paper: Aceso: Achieving Efficient Fault Tolerance in Memory-Disaggregated Key-Value …☆24Oct 20, 2024Updated last year
- An auxiliary project analysis of the characteristics of KV in DiT Attention.☆33Nov 29, 2024Updated last year
- ☆28Aug 14, 2024Updated last year