zasdfgbnm / autonvtx
Automatically insert nvtx ranges to PyTorch models
☆17Updated 3 years ago
Related projects ⓘ
Alternatives and complementary repositories for autonvtx
- Codebase associated with the PyTorch compiler tutorial☆44Updated 5 years ago
- ☆36Updated 5 months ago
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆26Updated last year
- Convert nvprof profiles into about:tracing compatible JSON files☆67Updated 3 years ago
- a high performance system for customized-precision distributed deep learning☆12Updated 3 years 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
- DLPack for Tensorflow☆36Updated 4 years ago
- Customized matrix multiplication kernels☆53Updated 2 years ago
- PyProf2: PyTorch Profiling tool☆83Updated 4 years ago
- A library for syntactically rewriting Python programs, pronounced (sinner).☆70Updated 2 years ago
- Python bindings for NVTX☆66Updated last year
- A logging tool for deep learning.☆52Updated 2 years ago
- ☆48Updated 8 months ago
- A tracing JIT for PyTorch☆17Updated 2 years ago
- ☆14Updated last month
- Training neural networks in TensorFlow 2.0 with 5x less memory☆129Updated 2 years ago
- This repository contains the results and code for the MLPerf™ Training v1.0 benchmark.☆37Updated 8 months ago
- Benchmarks to capture important workloads.☆28Updated 5 months ago
- CUDA templates for tile-sparse matrix multiplication based on CUTLASS.☆49Updated 6 years ago
- Test winograd convolution written in TVM for CUDA and AMDGPU☆40Updated 6 years ago
- nGraph™ Backend for ONNX☆42Updated last year
- oneCCL Bindings for Pytorch*☆86Updated 3 weeks ago
- MLPerf™ logging library☆30Updated last week
- A GPU performance profiling tool for PyTorch models☆22Updated 2 years ago
- Personal collection of references for high performance mixed precision training.☆41Updated 5 years ago
- Stride visualizations☆36Updated 6 years ago
- This repository contains the results and code for the MLPerf™ Training v0.7 benchmark.☆56Updated last year
- Worked example of the process from Python source to CUDA kernel execution with Numba☆36Updated 2 months ago
- ☆23Updated 9 months ago