Sparse multi-dimensional arrays for the PyData ecosystem

python numpy sparse-data sparse-matrix sparse-matrices sparse-arrays array-api

• Updated Nov 11, 2026
• Python

Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python

python data-science machine-learning hpc gpu numpy mpi parallel-computing distributed-computing pytorch scientific-computing data-analytics high-performance-computing tensors multi-gpu mpi4py array-api

• Updated Nov 12, 2026
• Python

Modular and scalable computational imaging in Python with GPU/out-of-core computing.

machine-learning modular scalable computer-vision image-reconstruction python-library image-processing artificial-intelligence bayesian-methods computational-imaging image-restoration image-enhancement sensing matrix-free sparse-learning inverse-problem foundation-models array-api python-hpc array-computing

• Updated Apr 2, 2025
• Python

Compatibility layer for common array libraries to support the Array API

numpy pytorch dask cupy array-api

• Updated Nov 9, 2026
• Python

Static typing support for the array API standard

python static-analysis arrays data-api static-typing array-api

• Updated Jun 27, 2026
• Python

Array APIs to write ONNX Graphs

machine-learning numpy onnx array-api

• Updated Jan 18, 2026
• Python

Modular, composable Sequential Monte Carlo framework

torch bayesian-methods sampling bayesian-inference sequential-monte-carlo cupy sampling-methods jax array-api

• Updated Oct 31, 2026
• Python

Automatic differentiation for the NumPy code you already write

python rust numpy automatic-differentiation scientific-computing autodiff differentiable-programming array-api

• Updated Oct 26, 2026
• Python

Pre-quaxed libraries for use in multiple dispatch over abstract array types in JAX.

array multiple-dispatch ducktyping jax array-api

• Updated Nov 5, 2026
• Python

Autogenerated types for array-api-compat and array API - IDE friendly. `pip install types-array-api` and ready to go!

python numpy typing pytorch mypy jax mypy-stubs type-stubs array-api array-api-typing

• Updated Nov 9, 2026
• Python

array-api based simple volume renderer

python dicom medical-imaging volume-rendering nifti medical-images mha my-projects array-api

• Updated Nov 11, 2026
• Python

Dimensioned Python arrays with dimension names and physical units

python units array-api

• Updated Nov 12, 2024
• Python

Vilenkin–Kuznetsov–Smorodinsky polyspherical (hyperspherical) coordinates in NumPy / PyTorch

hypersphere spherical-coordinates array-api

• Updated Nov 9, 2026
• Python

JIT decorator supporting multiple array API compatible libraries

jit pytorch numba cupy jax array-api

• Updated Nov 12, 2026
• Python

A list of awesome array API standard related projects. Feel free to add your project.

best-of array-api

• Updated Oct 30, 2026

Arbitrary precision floating / ball arithmetic (interval arithmetic) dtype in NumPy / array API

numpy arbitrary-precision flint acb interval-arithmetic arb mpmath array-api python-flint

• Updated Nov 8, 2026
• Python

Compute Jacobi polynomial from order 0 to N efficiently in NumPy / PyTorch / JAX / array API

numpy pytorch numba jax legendre-polynomials array-api jacobi-polynomials gegenbauer-polynomials numba-cuda

• Updated Nov 12, 2026
• Python

nanobind bindings for duckDB's C API.

jax duckdb polars array-api nanobind

• Updated Nov 11, 2026
• C++

Shift Nth row N steps in NumPy / PyTorch / TensorFlow / JAX

python numpy torch move jax manupulation advanced-indexing array-api

• Updated Nov 11, 2026
• Python

Cross-backend Python array library based on the Array API Standard.

numpy pytorch jax array-api

• Updated Nov 5, 2026
• Python