Array Api Python Packages

array-api-compat

Compatibility layer for common array libraries to support the Array API

sparse

Sparse multi-dimensional arrays for the PyData ecosystem

types-array-api

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

quaxed

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

jacobi-poly

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

ultrasphere

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

shift-nth-row-n-steps

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

heat

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

array-api

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

numpy-flint-arb

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

onnx-array-api

Array APIs to write ONNX Graphs

xbarray

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

pyxu

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

array-api-jax-compat

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

array-api-jit

JIT decorator supporting multiple array API compatible libraries

pydims

Dimensioned Python arrays with dimension names and physical units

pycsou

Library for solving linear inverse problems with state-of-the-art proximal algorithms.

array-api-typing

Coming soon - Static typing support for the array API standard