Scalable array processing with bounded memory

Data Parallel Extension for Numba

Abstract your array operations.

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

An implementation of chunked, compressed, N-dimensional arrays for Python.

NVIDIA Math Libraries for the Python Ecosystem

Composable transformations of Python+NumPy programs: differentiate, vectorize, JIT to GPU/TPU, and more

N-D labeled arrays and datasets in Python

Rewriting Finch in mlir

MLX: An array framework for Apple silicon