octoml / octoml-profile
Home for OctoML PyTorch Profiler
☆107Updated last year
Alternatives and similar repositories for octoml-profile:
Users that are interested in octoml-profile are comparing it to the libraries listed below
- A schedule language for large model training☆141Updated 5 months ago
- ☆141Updated 3 weeks ago
- Torch Distributed Experimental☆116Updated 4 months ago
- A library to analyze PyTorch traces.☆312Updated 2 weeks ago
- Applied AI experiments and examples for PyTorch☆182Updated last week
- extensible collectives library in triton☆73Updated 2 months ago
- Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …☆63Updated 2 years ago
- ☆44Updated last year
- ☆14Updated last week
- A performant, memory-efficient checkpointing library for PyTorch applications, designed with large, complex distributed workloads in mind…☆150Updated last week
- ☆68Updated last year
- ☆48Updated 9 months ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆281Updated this week
- ☆55Updated 6 months ago
- PyTorch RFCs (experimental)☆130Updated 3 months ago
- System for automated integration of deep learning backends.☆48Updated 2 years ago
- Training neural networks in TensorFlow 2.0 with 5x less memory☆130Updated 2 years ago
- Research and development for optimizing transformers☆125Updated 3 years ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆91Updated 5 months ago
- ☆59Updated last month
- MLIR-based partitioning system☆49Updated this week
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆177Updated this week
- NVIDIA Resiliency Extension is a python package for framework developers and users to implement fault-tolerant features. It improves the …☆53Updated 2 weeks ago
- Fast low-bit matmul kernels in Triton☆161Updated last week
- ☆150Updated 6 months ago
- ☆94Updated 3 months ago
- A library for syntactically rewriting Python programs, pronounced (sinner).☆70Updated 2 years ago
- Memory Optimizations for Deep Learning (ICML 2023)☆60Updated 9 months ago
- Benchmarks to capture important workloads.☆28Updated 6 months ago
- 🚀 Collection of components for development, training, tuning, and inference of foundation models leveraging PyTorch native components.☆175Updated this week