NVIDIA / nvmath-pythonLinks
NVIDIA Math Libraries for the Python Ecosystem
☆535Updated last week
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☆214Updated this week
- The Foundation for All Legate Libraries☆232Updated this week
- NumPy and SciPy on Multi-Node Multi-GPU systems☆952Updated this week
- Numbast is a tool to build an automated pipeline that converts CUDA APIs into Numba bindings.☆52Updated this week
- NVIDIA curated collection of educational resources related to general purpose GPU programming.☆889Updated this week
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆312Updated this week
- Data Parallel Extension for NumPy☆118Updated this week
- An efficient C++17 GPU numerical computing library with Python-like syntax☆1,365Updated last week
- JAX-Toolbox☆363Updated last week
- Fast and Furious AMD Kernels☆298Updated this week
- Data Parallel Extension for Numba☆87Updated 2 months ago
- Kernel Tuner☆372Updated 2 weeks ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆633Updated this week
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆473Updated 3 weeks ago
- High-Performance SGEMM on CUDA devices☆112Updated 10 months ago
- A plugin for Jupyter Notebook to run CUDA C/C++ code☆255Updated last year
- CUDA Kernel Benchmarking Library☆769Updated last week
- RAPIDS Memory Manager☆661Updated this week
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆498Updated last week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆362Updated this week
- A next generation Python CMake adaptor and Python API for plugins☆420Updated this week
- OpTree: Optimized PyTree Utilities☆199Updated last week
- Python SYCL bindings and SYCL-based Python Array API library☆118Updated this week
- A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python☆333Updated last year
- AutoBound automatically computes upper and lower bounds on functions.☆363Updated last month
- Official Problem Sets / Reference Kernels for the GPU MODE Leaderboard!☆160Updated 2 weeks ago
- An Online Deep Learning Interface for HPC programs on NVIDIA GPUs☆175Updated this week
- Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)☆455Updated 2 weeks ago
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆700Updated this week
- ☆598Updated last month