Search 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

Typing for array API and array-api-compat

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

heat

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

onnx-array-api

Array APIs to write ONNX Graphs

ultrasphere

Hyperspherical coordinates in NumPy / PyTorch / JAX.

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.

pyxu

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

pydims

Dimensioned Python arrays with dimension names and physical units

numpy-flint-arb

Arbitrary precision ball arithmetic (interval arithmetic) dtype in NumPy

array-api-jax-compat

Array-API JAX compatibility

array-api-jit

JIT decorator supporting multiple array API compatible libraries

pycsou

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

array-api-typing

Coming soon - Static typing support for the array API standard