Quansight / torch-buildLinks
Collection of scripts to build PyTorch and the domain libraries from source.
☆12Updated 2 weeks ago
Alternatives and similar repositories for torch-build
Users that are interested in torch-build are comparing it to the libraries listed below
Sorting:
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆68Updated 7 months ago
- An experimental CPU backend for Triton (https//github.com/openai/triton)☆47Updated 3 months ago
- NPBench - A Benchmarking Suite for High-Performance NumPy☆89Updated last month
- MLIR-based partitioning system☆148Updated this week
- TORCH_LOGS parser for PT2☆64Updated last week
- A lightweight triton-based General Matrix Multiplication (GEMM) library.☆27Updated last week
- ☆17Updated 3 years ago
- A lightweight, Pythonic, frontend for MLIR☆80Updated 2 years ago
- POC work on MLIR backend☆61Updated last year
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆308Updated this week
- Einsum optimization using opt_einsum and PyTorch FX graph rewriting☆22Updated 3 years ago
- ☆49Updated this week
- Ahead of Time (AOT) Triton Math Library☆83Updated last week
- MLPerf™ logging library☆37Updated last month
- NVIDIA's launch, startup, and logging scripts used by our MLPerf Training and HPC submissions☆32Updated 2 months ago
- Numbast is a tool to build an automated pipeline that converts CUDA APIs into Numba bindings.☆52Updated this week
- ☆28Updated last month
- Open source cross-platform compiler for compute-intensive loops used in AI algorithms, from Microsoft Research☆112Updated 2 years ago
- extensible collectives library in triton☆91Updated 7 months ago
- A tracing JIT compiler for PyTorch☆13Updated 3 years ago
- ☆50Updated last year
- The CUDA target for Numba☆210Updated this week
- ☆21Updated 8 months ago
- Bandwidth test for ROCm☆69Updated last week
- Matrix multiplication on GPUs for matrices stored on a CPU. Similar to cublasXt, but ported to both NVIDIA and AMD GPUs.☆32Updated 7 months ago
- Example of using pytorch's open device registration API☆31Updated 3 years ago
- A Data-Centric Compiler for Machine Learning☆85Updated last year
- Graph-indexed Pandas DataFrames for analyzing hierarchical performance data☆34Updated 2 weeks ago
- ☆49Updated 6 months ago
- LLM training in simple, raw C/CUDA☆108Updated last year