Performance benchmarking with ColossalAI
☆39Jul 6, 2022Updated 3 years ago
Alternatives and similar repositories for ColossalAI-Benchmark
Users that are interested in ColossalAI-Benchmark are comparing it to the libraries listed below
Sorting:
- Examples of training models with hybrid parallelism using ColossalAI☆339Mar 23, 2023Updated 2 years ago
- Sky Computing: Accelerating Geo-distributed Computing in Federated Learning☆90Nov 22, 2022Updated 3 years ago
- Scalable PaLM implementation of PyTorch☆190Dec 19, 2022Updated 3 years ago
- ☆30Sep 4, 2023Updated 2 years ago
- A Python library transfers PyTorch tensors between CPU and NVMe☆125Nov 27, 2024Updated last year
- A collection of models built with ColossalAI☆32Nov 22, 2022Updated 3 years ago
- ☆28Jul 11, 2021Updated 4 years ago
- A curated list of awesome projects and papers for distributed training or inference☆267Oct 8, 2024Updated last year
- Efficient Auto-scalable Scientific Infrastructure for Engineers and Researchers☆15Sep 8, 2025Updated 6 months ago
- A memory efficient DLRM training solution using ColossalAI☆107Nov 22, 2022Updated 3 years ago
- Large-scale model inference.☆627Sep 12, 2023Updated 2 years ago
- GPT Demo with hybrid distributed training☆10Dec 1, 2022Updated 3 years ago
- I have created a dataset of Image-Text-Pairs by using the cosine similarity of the CLIP embeddings of the image & it's caption derrived f…☆16Apr 22, 2021Updated 4 years ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆81Nov 19, 2024Updated last year
- Implementation of AAAI 2022 Paper: Go wider instead of deeper☆32Oct 27, 2022Updated 3 years ago
- ☆23Jan 7, 2022Updated 4 years ago
- High Performance Grouped GEMM in PyTorch☆30May 10, 2022Updated 3 years ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆222Aug 19, 2024Updated last year
- An interference-aware scheduler for fine-grained GPU sharing☆161Nov 26, 2025Updated 3 months ago
- Depict GPU memory footprint during DNN training of PyTorch☆11Nov 17, 2022Updated 3 years ago
- [ACL 2023] The official implementation of "CAME: Confidence-guided Adaptive Memory Optimization"☆97Mar 22, 2025Updated 11 months ago
- Slicing a PyTorch Tensor Into Parallel Shards☆300Jun 7, 2025Updated 9 months ago
- 💻 Terminal-Agent with Human-in-the-Loop Learning☆36Jan 16, 2026Updated 2 months ago
- TACOS: [T]opology-[A]ware [Co]llective Algorithm [S]ynthesizer for Distributed Machine Learning☆32Jun 13, 2025Updated 9 months ago
- Performance of the C++ interface of flash attention and flash attention v2 in large language model (LLM) inference scenarios.☆43Feb 27, 2025Updated last year
- The official implementation of "Helen: Optimizing CTR Prediction Models with Frequency-wise Hessian Eigenvalue Regularization"☆16Mar 14, 2024Updated 2 years ago
- Userspace eBPF Runtime Benchmarking Test Suite and Results☆16Mar 13, 2026Updated last week
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆164Jan 12, 2026Updated 2 months ago
- ☆47Dec 13, 2024Updated last year
- Efficient AI Inference & Serving☆480Jan 8, 2024Updated 2 years ago
- 面向多平台编译优化的深度学习中间表示☆10Oct 28, 2024Updated last year
- Drop-in library for tracking the memory allocations of CUDA applications☆14Nov 17, 2017Updated 8 years ago
- Lossless Training Speed Up by Unbiased Dynamic Data Pruning☆345Sep 24, 2024Updated last year
- Spack package repository maintained by Student Cluster Competition Team @ Sun Yat-sen University.☆16Aug 20, 2025Updated 7 months ago
- A unified programming framework for high and portable performance across FPGAs and GPUs☆11Mar 23, 2025Updated 11 months ago
- An MLIR-based AI compiler designed for Python frontend to RISC-V DSA☆13Oct 10, 2024Updated last year
- A tool to detect infrastructure issues on cloud native AI systems☆52Sep 18, 2025Updated 6 months ago
- Distributed Communication-Optimal Matrix-Matrix Multiplication Algorithm☆213Mar 3, 2026Updated 2 weeks ago
- Models and examples built with OneFlow☆101Oct 16, 2024Updated last year