ruipeterpan / torch_profilerLinks
Simple PyTorch profiler that combines DeepSpeed Flops Profiler and TorchInfo
☆11Updated 2 years ago
Alternatives and similar repositories for torch_profiler
Users that are interested in torch_profiler are comparing it to the libraries listed below
Sorting:
- PyTorch compilation tutorial covering TorchScript, torch.fx, and Slapo☆17Updated 2 years ago
- Official resporitory for "IPDPS' 24 QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devices".☆20Updated last year
- An Attention Superoptimizer☆22Updated 10 months ago
- SOTA Learning-augmented Systems☆37Updated 3 years ago
- An external memory allocator example for PyTorch.☆16Updated 3 months ago
- ☆25Updated 2 years ago
- TiledLower is a Dataflow Analysis and Codegen Framework written in Rust.☆14Updated last year
- Artifact for "Shockwave: Fair and Efficient Cluster Scheduling for Dynamic Adaptation in Machine Learning" [NSDI '23]☆46Updated 3 years ago
- ☆21Updated 3 years ago
- Artifact for "Apparate: Rethinking Early Exits to Tame Latency-Throughput Tensions in ML Serving" [SOSP '24]☆25Updated last year
- DISB is a new DNN inference serving benchmark with diverse workloads and models, as well as real-world traces.☆57Updated last year
- A source-to-source compiler for optimizing CUDA dynamic parallelism by aggregating launches☆15Updated 6 years ago
- ☆10Updated 3 years ago
- A Cluster-Wide Model Manager to Accelerate DNN Training via Automated Training Warmup☆35Updated 2 years ago
- ☆13Updated 2 years ago
- ☆15Updated 3 years ago
- AI model training on heterogeneous, geo-distributed resources☆21Updated this week
- ☆17Updated 2 years ago
- A GPU-accelerated DNN inference serving system that supports instant kernel preemption and biased concurrent execution in GPU scheduling.☆43Updated 3 years ago
- Deferred Continuous Batching in Resource-Efficient Large Language Model Serving (EuroMLSys 2024)☆19Updated last year
- RL-Scope: Cross-Stack Profiling for Deep Reinforcement Learning Workloads☆46Updated 4 years ago
- Paper list for accleration of transformers☆13Updated 2 years ago
- ☆14Updated 3 weeks ago
- ☆30Updated last month
- ☆44Updated 4 years ago
- ☆14Updated 3 years ago
- This is the proof-of-concept CPU implementation of ASPEN used for the NeurIPS'23 paper ASPEN: Breaking Operator Barriers for Efficient Pa…☆13Updated last year
- ☆19Updated last year
- ☆19Updated 5 months ago
- ☆68Updated 2 years ago