gmarkall / life-of-a-numba-kernel
Worked example of the process from Python source to CUDA kernel execution with Numba
☆40Updated 7 months ago
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
- An Aspiring Drop-In Replacement for Pandas at Scale☆75Updated 3 years ago
- Collection of scripts to build PyTorch and the domain libraries from source.☆10Updated last week
- Numbast is a tool to build an automated pipeline that converts CUDA APIs into Numba bindings.☆43Updated this week
- The CUDA target for Numba☆102Updated this week
- Automatically insert nvtx ranges to PyTorch models☆17Updated 3 years ago
- ☆16Updated 2 years ago
- The Foundation for All Legate Libraries☆212Updated this week
- Einsum optimization using opt_einsum and PyTorch FX graph rewriting☆21Updated 3 years ago
- ☆51Updated 8 months ago
- NPBench - A Benchmarking Suite for High-Performance NumPy☆80Updated 2 weeks ago
- A tracing JIT compiler for PyTorch☆13Updated 3 years ago
- MLPerf™ logging library☆34Updated last week
- Material for the SC22 Deep Learning at Scale Tutorial☆41Updated last year
- Codebase associated with the PyTorch compiler tutorial☆45Updated 5 years ago
- Test suite for Python array API standard compliance☆67Updated this week
- A library that translates Python and NumPy to optimized distributed systems code.☆132Updated 2 years ago
- ☆21Updated last month
- ☆16Updated 6 months ago
- No-GIL Python environment featuring NVIDIA Deep Learning libraries.☆53Updated this 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
- CUDA templates for tile-sparse matrix multiplication based on CUTLASS.☆51Updated 7 years ago
- Home for OctoML PyTorch Profiler☆112Updated last year
- Benchmarks to capture important workloads.☆31Updated 2 months ago
- ☆163Updated 10 months ago
- ArrayViews: creating specific views to array storage objects☆17Updated 6 years ago
- Dive into Jax, Flax, XLA and C++☆31Updated 5 years ago
- Customized matrix multiplication kernels☆54Updated 3 years ago
- RFC document, tooling and other content related to the array API standard☆233Updated 2 weeks ago
- A benchmark to measure performance of popular Gradient boosting algorithms against popular ML datasets.☆38Updated 2 years ago
- POC work on MLIR backend☆54Updated 7 months ago