Memmap

Create a memory-map to an array stored in a binary file on disk.

Abstract Signature:

Memmap(filename: str, dtype: str = uint8, mode: str = r+, shape: Tuple[int])

PyTorchJAX (Core)NumPyKerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: —

Strategy: Plugin (numpy_memmap_to_torch)

JAX (Core)

API: jax.numpy.memmap

Strategy: Direct Mapping

Official Docs ↗

NumPy

API: numpy.memmap

Strategy: Direct Mapping

Official Docs ↗

Keras

API: —

Strategy: Plugin (numpy_fallback)

TensorFlow

API: —

Strategy: Plugin (numpy_fallback)

Apple MLX

API: —

Strategy: Plugin (numpy_fallback)

Flax NNX

API: jax.numpy.memmap

Strategy: Direct Mapping

Official Docs ↗

PaxML / Praxis

API: jax.numpy.memmap

Strategy: Direct Mapping

Official Docs ↗