zasdfgbnm / autonvtx
Automatically insert nvtx ranges to PyTorch models
☆17Updated 3 years ago
Alternatives and similar repositories for autonvtx:
Users that are interested in autonvtx are comparing it to the libraries listed below
- Codebase associated with the PyTorch compiler tutorial☆46Updated 5 years ago
- Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …☆64Updated 3 years ago
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆26Updated last year
- PyProf2: PyTorch Profiling tool☆82Updated 4 years ago
- ☆12Updated 3 years ago
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆43Updated 2 weeks ago
- Collection of scripts to build PyTorch and the domain libraries from source.☆10Updated this week
- pytest plugin for a better developer experience when working with the PyTorch test suite☆44Updated 3 years ago
- DLPack for Tensorflow☆36Updated 4 years ago
- Customized matrix multiplication kernels☆53Updated 3 years ago
- Convert nvprof profiles into about:tracing compatible JSON files☆69Updated 3 years ago
- ☆38Updated 3 months ago
- A library for syntactically rewriting Python programs, pronounced (sinner).☆70Updated 3 years ago
- Example python package with pybind11 cpp extension☆57Updated 4 years ago
- Einsum optimization using opt_einsum and PyTorch FX graph rewriting☆20Updated 3 years ago
- Test winograd convolution written in TVM for CUDA and AMDGPU☆40Updated 6 years ago
- ☆49Updated last year
- CuPy Benchmark☆12Updated 5 years ago
- A tracing JIT compiler for PyTorch☆13Updated 3 years ago
- benchmarking some transformer deployments☆26Updated last year
- Hacks for PyTorch☆18Updated last year
- torch::deploy (multipy for non-torch uses) is a system that lets you get around the GIL problem by running multiple Python interpreters i…☆179Updated 3 months ago
- PyTorch RFCs (experimental)☆130Updated 6 months ago
- Python bindings for NVTX☆66Updated last year
- Build TVM docker image for production compilation deployments☆13Updated 3 years ago
- a high performance system for customized-precision distributed deep learning☆12Updated 4 years ago
- Home for OctoML PyTorch Profiler☆108Updated last year
- Distributed ML Optimizer☆30Updated 3 years ago
- Python Binding to NVRTC☆79Updated 5 months ago
- A Ray-based data loader with per-epoch shuffling and configurable pipelining, for shuffling and loading training data for distributed tra…☆18Updated 2 years ago