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.

onnx-array-api

Array APIs to write ONNX Graphs

jacobi-poly

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

heat

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

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

array-api

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

xbarray

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

array-api-jax-compat

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

pyxu

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

numpy-flint-arb

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

pydims

Dimensioned Python arrays with dimension names and physical units

array-api-jit

JIT decorator supporting multiple array API compatible libraries

array-api-typing

Static typing support for the array API standard

pycsou

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