Quansight / torch-build
Collection of scripts to build PyTorch and the domain libraries from source.
☆10Updated last week
Alternatives and similar repositories for torch-build:
Users that are interested in torch-build are comparing it to the libraries listed below
- NVIDIA's launch, startup, and logging scripts used by our MLPerf Training and HPC submissions☆26Updated this week
- The CUDA target for Numba☆102Updated this week
- Worked example of the process from Python source to CUDA kernel execution with Numba☆40Updated 7 months ago
- A tracing JIT compiler for PyTorch☆13Updated 3 years ago
- Numbast is a tool to build an automated pipeline that converts CUDA APIs into Numba bindings.☆43Updated this week
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆53Updated this week
- Einsum optimization using opt_einsum and PyTorch FX graph rewriting☆21Updated 3 years ago
- NPBench - A Benchmarking Suite for High-Performance NumPy☆80Updated 2 weeks ago
- ☆21Updated last month
- Automatically insert nvtx ranges to PyTorch models☆17Updated 3 years ago
- ☆16Updated 2 years ago
- POC work on MLIR backend☆54Updated 7 months ago
- MLPerf™ logging library☆34Updated last week
- An experimental CPU backend for Triton (https//github.com/openai/triton)☆40Updated last month
- pytorch ucc plugin☆21Updated 3 years ago
- An Aspiring Drop-In Replacement for Pandas at Scale☆75Updated 3 years ago
- End to End steps for adding custom ops in PyTorch.☆21Updated 4 years ago
- Random number library that generate pseudo-random and quasi-random numbers.☆26Updated this week
- A tracing JIT for PyTorch☆17Updated 2 years ago
- ☆51Updated 8 months ago
- Bandwidth test for ROCm☆54Updated this week
- A lightweight, Pythonic, frontend for MLIR☆81Updated last year
- Analyze graph/hierarchical performance data using pandas dataframes☆113Updated 2 months ago
- CUDA Templates for Linear Algebra Subroutines☆20Updated last week
- ☆29Updated this week
- MLIR-based partitioning system☆80Updated this week
- Data Parallel Extension for NumPy☆105Updated this week
- ☆13Updated 3 years ago
- Python bindings for OpenSHMEM☆16Updated this week
- Open source cross-platform compiler for compute-intensive loops used in AI algorithms, from Microsoft Research☆110Updated last year