ruipeterpan / torch_profiler
Simple PyTorch profiler that combines DeepSpeed Flops Profiler and TorchInfo
☆9Updated last year
Alternatives and similar repositories for torch_profiler:
Users that are interested in torch_profiler are comparing it to the libraries listed below
- PyTorch compilation tutorial covering TorchScript, torch.fx, and Slapo☆19Updated last year
- SOTA Learning-augmented Systems☆34Updated 2 years ago
- Official resporitory for "IPDPS' 24 QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devices".☆19Updated 10 months ago
- An Attention Superoptimizer☆20Updated 8 months ago
- Artifact for "Apparate: Rethinking Early Exits to Tame Latency-Throughput Tensions in ML Serving" [SOSP '24]☆22Updated last month
- An external memory allocator example for PyTorch.☆14Updated 3 years ago
- Tacker: Tensor-CUDA Core Kernel Fusion for Improving the GPU Utilization while Ensuring QoS☆18Updated 3 years ago
- A Cluster-Wide Model Manager to Accelerate DNN Training via Automated Training Warmup☆34Updated 2 years ago
- ☆18Updated 2 years ago
- ☆22Updated 4 years ago
- A Sparse-tensor Communication Framework for Distributed Deep Learning☆13Updated 3 years ago
- [IJCAI2023] An automated parallel training system that combines the advantages from both data and model parallelism. If you have any inte…☆51Updated last year
- Code associated with the paper **Fine-tuning Language Models over Slow Networks using Activation Compression with Guarantees**.☆27Updated last year
- MobiSys#114☆21Updated last year
- ☆24Updated last year
- ☆16Updated 2 years ago
- ☆39Updated this week
- CS/ECE/ME/EP 759 (High Performance Computing for Engineering Applications) Course Project: Cautiously Aggressive GPU Space Sharing to Imp…☆8Updated 3 years ago
- [ICDCS 2023] DeAR: Accelerating Distributed Deep Learning with Fine-Grained All-Reduce Pipelining☆12Updated last year
- Code for "Shockwave: Fair and Efficient Cluster Scheduling for Dynamic Adaptation in Machine Learning" [NSDI '23]☆39Updated 2 years ago
- Experimental deep learning framework written in Rust☆14Updated 2 years ago
- Code for Large Graph Convolutional Network Training with GPU-Oriented Data Communication Architecture (accepted by PVLDB).The outdated wr…☆8Updated last year
- Benchmark tests supporting the TiledCUDA library.☆12Updated 2 months ago
- website for systems seminar at UIUC☆17Updated this week
- General system research material (not limited to paper) reading notes.☆21Updated 3 years ago
- Artifact of ASPLOS'23 paper entitled: GRACE: A Scalable Graph-Based Approach to Accelerating Recommendation Model Inference☆17Updated last year
- Multi-Instance-GPU profiling tool☆56Updated last year
- (NeurIPS 2022) Automatically finding good model-parallel strategies, especially for complex models and clusters.☆37Updated 2 years ago
- Vector search with bounded performance.☆34Updated 11 months ago
- Personal blog + reading notes on system-ish papers☆15Updated last year