Quansight / torch-build
Collection of scripts to build PyTorch and the domain libraries from source.
☆10Updated this week
Alternatives and similar repositories for torch-build:
Users that are interested in torch-build are comparing it to the libraries listed below
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆59Updated 3 weeks ago
- NPBench - A Benchmarking Suite for High-Performance NumPy☆81Updated 2 weeks ago
- A tracing JIT compiler for PyTorch☆13Updated 3 years ago
- ☆21Updated 2 months ago
- Worked example of the process from Python source to CUDA kernel execution with Numba☆40Updated 7 months ago
- POC work on MLIR backend☆55Updated 8 months ago
- NVIDIA's launch, startup, and logging scripts used by our MLPerf Training and HPC submissions☆26Updated this week
- A lightweight, Pythonic, frontend for MLIR☆81Updated last year
- The CUDA target for Numba☆112Updated this week
- ☆51Updated 9 months ago
- Numbast is a tool to build an automated pipeline that converts CUDA APIs into Numba bindings.☆44Updated this week
- An experimental CPU backend for Triton (https//github.com/openai/triton)☆40Updated last month
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆26Updated 2 years ago
- MLIR-based partitioning system☆82Updated this week
- ☆50Updated last year
- TORCH_LOGS parser for PT2☆37Updated 2 weeks ago
- Benchmarks to capture important workloads.☆31Updated 3 months ago
- ☆13Updated 3 years ago
- ☆23Updated last year
- Open source cross-platform compiler for compute-intensive loops used in AI algorithms, from Microsoft Research☆109Updated last year
- MLPerf™ logging library☆36Updated 3 weeks ago
- An IR for efficiently simulating distributed ML computation.☆28Updated last year
- ☆31Updated last week
- Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …☆65Updated 3 years ago
- extensible collectives library in triton☆86Updated last month
- Notes and artifacts from the ONNX steering committee☆26Updated last week
- ☆16Updated 7 months ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆84Updated last week
- Prototype routines for GPU quantization written using PyTorch.☆21Updated 2 months ago
- Bandwidth test for ROCm☆54Updated this week