HIPS / autograd

Efficiently computes derivatives of NumPy code.

pydata / xarray

N-D labeled arrays and datasets in Python

zarr-developers / zarr-python

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

inducer / pycuda

CUDA integration for Python, plus shiny features

inducer / pyopencl

OpenCL integration for Python, plus shiny features

K3D-tools / K3D-jupyter

WebGL 3D plotting for Jupyter: meshes, isosurfaces, volumes, voxels and point clouds from NumPy arrays, with a path-traced cinematic renderer.

ilanschnell / bitarray

efficient arrays of booleans for Python

pydata / sparse

Sparse multi-dimensional arrays for the PyData ecosystem

pymc-devs / pytensor

PyTensor allows you to define, optimize, and efficiently evaluate mathematical expressions involving multi-dimensional arrays.

inducer / loopy

A code generator for array-based code on CPUs and GPUs

mpi4jax / mpi4jax

Zero-copy MPI communication of JAX arrays, for turbo-charged HPC applications in Python :zap: