OmniNumPy is an experimental compatibility layer that sits on top of modern NumPy (≥2.0) and lets you run the same code across multiple array backends — NumPy, PyTorch, CuPy, and JAX — with minimal changes.
python legacy gpu numpy torch compatibility scientific-computing shim cupy compatibility-testing numpy-library compatibility-layer jax numpy-neural-network numpy-python numpy-compatibility numpy-2x
-
Updated
Sep 30, 2025 - Python