Array Api Python Packages

array-api-compat

Compatibility layer for common array libraries to support the Array API

2M 132 50

sparse

Sparse multi-dimensional arrays for the PyData ecosystem

1.1M 669 147

quaxed

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

18K 8 3

advect

Automatic differentiation for the NumPy code you already write

11K 9 1

types-array-api

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

7K 7 1

heat

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

2K 251 72

jacobi-poly

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

2K 1 0

ultrasphere

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

1K 2 0

shift-nth-row-n-steps

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

1K 1 0

onnx-array-api

Array APIs to write ONNX Graphs

658 11 0

xbarray

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

426 0 0

array-api-jax-compat

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

386 8 3

array-api

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

275 7 1

pydims

Dimensioned Python arrays with dimension names and physical units

191 2 0

pyxu

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

188 142 17

numpy-flint-arb

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

185 1 1

pycsou

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

173 142 17

array-api-jit

JIT decorator supporting multiple array API compatible libraries

144 2 0

array-api-typing

Static typing support for the array API standard

80 27 6