NVIDIA / nvmath-pythonLinks
NVIDIA Math Libraries for the Python Ecosystem
☆544Updated 3 weeks ago
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☆251Updated this week
- The Foundation for All Legate Libraries☆233Updated this week
- Numbast is a tool to build an automated pipeline that converts CUDA APIs into Numba bindings.☆56Updated this week
- NumPy and SciPy on Multi-Node Multi-GPU systems☆965Updated last week
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆328Updated last week
- Data Parallel Extension for NumPy☆121Updated this week
- NVIDIA curated collection of educational resources related to general purpose GPU programming.☆1,146Updated this week
- Kernel Tuner☆381Updated last week
- Data Parallel Extension for Numba☆90Updated 4 months ago
- Fast and Furious AMD Kernels☆348Updated 2 weeks ago
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆507Updated last week
- An efficient C++20 GPU numerical computing library with Python-like syntax☆1,402Updated this week
- CUDA Tile IR is an MLIR-based intermediate representation and compiler infrastructure for CUDA kernel optimization, focusing on tile-base…☆823Updated 3 weeks ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆512Updated 2 weeks ago
- A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.☆739Updated this week
- cuTile is a programming model for writing parallel kernels for NVIDIA GPUs☆1,903Updated this week
- JAX-Toolbox☆382Updated this week
- Python SYCL bindings and SYCL-based Python Array API library☆121Updated last week
- LLM training in simple, raw C/CUDA☆112Updated last year
- Tilus is a tile-level kernel programming language with explicit control over shared memory and registers.☆440Updated last month
- Helpful kernel tutorials and examples for tile-based GPU programming☆630Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆380Updated this week
- This repository contains examples CUDA usage in Cython code.☆26Updated 4 years ago
- CUDA Kernel Benchmarking Library☆806Updated last week
- An Online Deep Learning Interface for HPC programs on NVIDIA GPUs☆183Updated last month
- A plugin for Jupyter Notebook to run CUDA C/C++ code☆257Updated last year
- Legate Sparse is a Legate library that aims to provide a distributed and accelerated drop-in replacement for the scipy.sparse library on …☆24Updated 2 weeks ago
- Extending JAX with custom C++ and CUDA code☆403Updated last year
- CUDA-L2: Surpassing cuBLAS Performance for Matrix Multiplication through Reinforcement Learning☆417Updated 3 weeks ago
- RAPIDS Memory Manager☆681Updated this week