NVIDIA / nvmath-pythonLinks
NVIDIA Math Libraries for the Python Ecosystem
☆542Updated last month
Alternatives and similar repositories for nvmath-python
Users that are interested in nvmath-python are comparing it to the libraries listed below
Sorting:
- The CUDA target for Numba☆233Updated this week
- Numbast is a tool to build an automated pipeline that converts CUDA APIs into Numba bindings.☆55Updated this week
- The Foundation for All Legate Libraries☆233Updated this week
- NumPy and SciPy on Multi-Node Multi-GPU systems☆964Updated this week
- NVIDIA curated collection of educational resources related to general purpose GPU programming.☆1,018Updated 2 weeks ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆320Updated this week
- Data Parallel Extension for NumPy☆118Updated this week
- Data Parallel Extension for Numba☆88Updated 2 months ago
- An efficient C++20 GPU numerical computing library with Python-like syntax☆1,373Updated this week
- Fast and Furious AMD Kernels☆324Updated this week
- Kernel Tuner☆377Updated this week
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆488Updated last week
- An Online Deep Learning Interface for HPC programs on NVIDIA GPUs☆177Updated 2 weeks ago
- cuTile is a programming model for writing parallel kernels for NVIDIA GPUs☆1,632Updated this week
- CUDA Kernel Benchmarking Library☆782Updated 2 weeks ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆691Updated this week
- LLM training in simple, raw C/CUDA☆108Updated last year
- Python SYCL bindings and SYCL-based Python Array API library☆120Updated last week
- RAPIDS Memory Manager☆666Updated this week
- A plugin for Jupyter Notebook to run CUDA C/C++ code☆257Updated last year
- Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)☆459Updated 2 weeks ago
- High-Performance SGEMM on CUDA devices☆113Updated 11 months ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆367Updated this week
- Meta-GPU lesson covering general aspects of GPU programming as well as specific frameworks☆98Updated 3 weeks ago
- Examples demonstrating available options to program multiple GPUs in a single node or a cluster☆843Updated 3 months ago
- JAX-Toolbox☆368Updated last week
- Helpful kernel tutorials and examples for tile-based GPU programming☆456Updated this week
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆433Updated last week
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆506Updated this week
- KvikIO - High Performance File IO☆233Updated this week