array-api

There are 13 repositories under array-api topic.

pydata/sparse
Sparse multi-dimensional arrays for the PyData ecosystem
Language:Python644 21 323131
helmholtz-analytics/heat
Distributed tensors and Machine Learning framework with GPU and MPI acceleration in Python
Language:Python223 7 78756
pyxu-org/pyxu
Modular and scalable computational imaging in Python with GPU/out-of-core computing.
Language:Python138 5 1416
data-apis/array-api-compat
Compatibility layer for common array libraries to support the Array API
Language:Python102 19 11837
data-apis/array-api-typing
Static typing support for the array API standard
Language:Python21 9 175
sdpython/onnx-array-api
Array APIs to write ONNX Graphs
Language:Python11 3 10
GalacticDynamics/quaxed
Pre-quaxed libraries for use in multiple dispatch over abstract array types in JAX.
Language:Python5 1 131
34j/types-array-api
Autogenerated types for array-api-compat and array API - IDE friendly. `pip install types-array-api` and ready to go!
Language:Python4 0 41
ar90n/vanilla-roll
array-api based simple volume renderer
Language:Python4 2 10
pydims/pydims
Dimensioned Python arrays with dimension names and physical units
Language:Python2 1 40
34j/shift-nth-row-n-steps
Shift Nth row N steps in NumPy / PyTorch / TensorFlow / JAX
Language:Python11
ultrasphere-dev/ultrasphere
Hyperspherical coordinates in NumPy / PyTorch
Language:Python1
34j/array-api-jit
JIT decorator supporting multiple array API compatible libraries
Language:Python0 1