nv-legate / cupynumeric
An Aspiring Drop-In Replacement for NumPy at Scale
☆828Updated this week
Alternatives and similar repositories for cupynumeric:
Users that are interested in cupynumeric are comparing it to the libraries listed below
- The Foundation for All Legate Libraries☆205Updated this week
- CUDA Python: Performance meets Productivity☆1,116Updated this week
- common in-memory tensor structure☆958Updated 2 weeks ago
- ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.☆887Updated 2 weeks ago
- NVIDIA Math Libraries for the Python Ecosystem☆236Updated 2 months ago
- An efficient C++17 GPU numerical computing library with Python-like syntax☆1,267Updated this week
- functorch is JAX-like composable function transforms for PyTorch.☆1,415Updated this week
- Examples demonstrating available options to program multiple GPUs in a single node or a cluster☆631Updated 2 weeks ago
- A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python☆317Updated 5 months ago
- Nvidia contributed CUDA tutorial for Numba☆249Updated 2 years ago
- RFC document, tooling and other content related to the array API standard☆228Updated last week
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,034Updated 10 months ago
- Utilities for Dask and CUDA interactions☆300Updated last week
- Distributed multigrid linear solver library on GPU☆529Updated last month
- ☆518Updated this week
- CUDA Kernel Benchmarking Library☆584Updated 3 months ago
- RAPIDS Memory Manager☆547Updated this week
- High performance model preprocessing library on PyTorch☆651Updated 11 months ago
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,076Updated last week
- A code generator for array-based code on CPUs and GPUs☆598Updated this week
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆468Updated this week
- Manipulate JSON-like data with NumPy-like idioms.☆868Updated this week
- PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code☆696Updated last year
- A GPU performance profiling tool for PyTorch models☆504Updated 3 years ago
- Kernel Tuner☆321Updated this week
- Python bindings for MPI☆837Updated this week
- Optax is a gradient processing and optimization library for JAX.☆1,815Updated this week
- Library for reading and writing large multi-dimensional arrays.☆1,385Updated this week
- High-performance automatic differentiation of LLVM and MLIR.☆1,348Updated this week
- Extending JAX with custom C++ and CUDA code☆387Updated 6 months ago