nv-legate / cupynumeric
An Aspiring Drop-In Replacement for NumPy at Scale
☆821Updated 3 weeks ago
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☆202Updated last month
- common in-memory tensor structure☆930Updated 4 months ago
- Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python☆457Updated last week
- ⚡️Optimizing einsum functions in NumPy, Tensorflow, Dask, and more with contraction order optimization.☆878Updated 3 months ago
- A suite of benchmarks for CPU and GPU performance of the most popular high-performance libraries for Python☆315Updated 3 months ago
- CUDA Python: Performance meets Productivity☆1,067Updated this week
- functorch is JAX-like composable function transforms for PyTorch.☆1,405Updated this week
- NVIDIA Math Libraries for the Python Ecosystem☆222Updated last month
- A Python-level JIT compiler designed to make unmodified PyTorch programs faster.☆1,019Updated 9 months ago
- High performance model preprocessing library on PyTorch☆651Updated 10 months ago
- RAPIDS Memory Manager☆525Updated this week
- KErnel OPerationS, on CPUs and GPUs, with autodiff and without memory overflows☆1,068Updated this week
- High-performance automatic differentiation of LLVM and MLIR.☆1,325Updated this week
- Hardware accelerated, batchable and differentiable optimizers in JAX.☆943Updated 4 months ago
- ☆506Updated this week
- Nvidia contributed CUDA tutorial for Numba☆240Updated 2 years ago
- CUDA integration for Python, plus shiny features☆1,898Updated 2 months ago
- JAX-Toolbox☆279Updated this week
- Optax is a gradient processing and optimization library for JAX.☆1,775Updated this week
- Python bindings for MPI☆829Updated this week
- A code generator for array-based code on CPUs and GPUs☆596Updated this week
- Manipulate JSON-like data with NumPy-like idioms.☆858Updated this week
- Utilities for Dask and CUDA interactions☆296Updated this week
- Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/☆2,212Updated this week
- Type annotations and runtime checking for shape and dtype of JAX/NumPy/PyTorch/etc. arrays. https://docs.kidger.site/jaxtyping/☆1,307Updated this week
- CUDA Kernel Benchmarking Library☆550Updated 2 months ago
- RFC document, tooling and other content related to the array API standard☆225Updated last week
- PyTorch, TensorFlow, JAX and NumPy — all of them natively using the same code☆695Updated last year
- Distributed multigrid linear solver library on GPU☆519Updated 5 months ago
- An open-source efficient deep learning framework/compiler, written in python.☆672Updated last week