Save

Saves an object (tensor or model) to a disk file.

Abstract Signature:

Save(obj, f: Union[str, Path], pickle_module, pickle_protocol: int = 2)

PyTorch

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

JAX (Core)

API: jax.numpy.save
Strategy: Plugin (io_handler)

NumPy

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

Keras

API: keras.saving.save_model
Strategy: Plugin (io_handler)

TensorFlow

API: tf.io.write_file
Strategy: Plugin (io_handler)

Apple MLX

API: mlx.core.save
Strategy: Direct Mapping
Official Docs ↗

Flax NNX

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

PaxML / Praxis

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