Save

Save an array to a binary file in NumPy .npy format.

Abstract Signature:

Save(file: str | Any, arr: Tensor, allow_pickle: bool = True)

PyTorch

API: torch.save
Strategy: Direct Mapping
Official Docs ↗

JAX (Core)

API: jax.numpy.save
Strategy: Direct Mapping
Official Docs ↗

NumPy

API: numpy.save
Strategy: Direct Mapping
Official Docs ↗

TensorFlow

API: save
Strategy: Plugin (io_handler)

Apple MLX

API: save
Strategy: Plugin (io_handler)

Flax NNX

API: jax.numpy.save
Strategy: Direct Mapping
Official Docs ↗