Quansight / torch-buildLinks
Collection of scripts to build PyTorch and the domain libraries from source.
☆12Updated last month
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.☆69Updated 7 months ago
- NPBench - A Benchmarking Suite for High-Performance NumPy☆89Updated 2 months ago
- MLIR-based partitioning system☆151Updated this week
- An experimental CPU backend for Triton (https//github.com/openai/triton)☆47Updated 3 months ago
- A tracing JIT compiler for PyTorch☆13Updated 4 years ago
- POC work on MLIR backend☆61Updated last year
- A lightweight triton-based General Matrix Multiplication (GEMM) library.☆28Updated last week
- Python bindings for UCX☆140Updated 2 months ago
- Numbast is a tool to build an automated pipeline that converts CUDA APIs into Numba bindings.☆53Updated this week
- ☆21Updated 9 months ago
- Einsum optimization using opt_einsum and PyTorch FX graph rewriting☆22Updated 3 years ago
- Benchmarks to capture important workloads.☆31Updated 10 months ago
- An Aspiring Drop-In Replacement for Pandas at Scale☆74Updated 4 years ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆319Updated this week
- A lightweight, Pythonic, frontend for MLIR☆80Updated 2 years ago
- ☆52Updated this week
- Data Parallel Extension for NumPy☆118Updated last week
- A Data-Centric Compiler for Machine Learning☆85Updated last year
- Graph-indexed Pandas DataFrames for analyzing hierarchical performance data☆34Updated 3 weeks ago
- MLPerf™ logging library☆37Updated last week
- Bandwidth test for ROCm☆70Updated last week
- TORCH_LOGS parser for PT2☆69Updated last month
- Open source cross-platform compiler for compute-intensive loops used in AI algorithms, from Microsoft Research☆113Updated 2 years ago
- NVIDIA's launch, startup, and logging scripts used by our MLPerf Training and HPC submissions☆35Updated 2 months ago
- Ahead of Time (AOT) Triton Math Library☆84Updated last month
- extensible collectives library in triton☆91Updated 8 months ago
- Matrix multiplication on GPUs for matrices stored on a CPU. Similar to cublasXt, but ported to both NVIDIA and AMD GPUs.☆32Updated 8 months ago
- ☆17Updated 3 years ago
- ☆50Updated last year
- ☆28Updated last week