gmarkall / life-of-a-numba-kernelLinks
Worked example of the process from Python source to CUDA kernel execution with Numba
☆42Updated last year
Alternatives and similar repositories for life-of-a-numba-kernel
Users that are interested in life-of-a-numba-kernel are comparing it to the libraries listed below
Sorting:
- An Aspiring Drop-In Replacement for Pandas at Scale☆74Updated 3 years ago
- A stand-alone implementation of several NumPy dtype extensions used in machine learning.☆301Updated last week
- The Foundation for All Legate Libraries☆228Updated this week
- A library that translates Python and NumPy to optimized distributed systems code.☆132Updated 3 years ago
- NPBench - A Benchmarking Suite for High-Performance NumPy☆89Updated last week
- A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python☆332Updated last year
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆64Updated 6 months ago
- Automatically insert nvtx ranges to PyTorch models☆19Updated 4 years ago
- PyTorch interface for the IPU☆181Updated 2 years ago
- RFC document, tooling and other content related to the array API standard☆257Updated last month
- Numbast is a tool to build an automated pipeline that converts CUDA APIs into Numba bindings.☆51Updated last week
- Customized matrix multiplication kernels☆57Updated 3 years ago
- Tensors and Dynamic neural networks in Python with strong GPU acceleration☆26Updated 2 years ago
- ☆177Updated last year
- PyTorch RFCs (experimental)☆135Updated 4 months ago
- Material for the SC22 Deep Learning at Scale Tutorial☆41Updated 2 years ago
- ☆52Updated last year
- Exploring using stdpar and Cython☆34Updated 4 years ago
- Dive into Jax, Flax, XLA and C++☆32Updated 5 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Nvidia contributed CUDA tutorial for Numba☆257Updated 3 years ago
- Collection of scripts to build PyTorch and the domain libraries from source.☆12Updated last month
- Introduction to CUDA programming☆126Updated 8 years ago
- A Python library for defining structured command-line flags.☆32Updated 3 months ago
- Documentation:☆124Updated 2 years ago
- A tensor-aware point-to-point communication primitive for machine learning☆273Updated 2 months ago
- Example python package with pybind11 cpp extension☆57Updated 4 years ago
- TorchFix - a linter for PyTorch-using code with autofix support☆148Updated last month
- Extending JAX with custom C++ and CUDA code☆401Updated last year
- TORCH_LOGS parser for PT2☆62Updated 3 weeks ago