SaveSafetensors

Save arrays to Safetensors format.

Abstract Signature:

SaveSafetensors(file: str, arrays: Dict[str, Tensor], metadata: Dict[str, str] | None)

PyTorchJAX (Core)KerasTensorFlowApple MLXFlax NNXPaxML / Praxis

PyTorch

API: —

Strategy: Macro 'save_file({arrays}, {file}, metadata={metadata})'

JAX (Core)

API: —

Strategy: Macro 'save_file({arrays}, {file}, metadata={metadata})'

Keras

API: —

Strategy: Plugin (save_safetensors_adapter)

TensorFlow

API: —

Strategy: Plugin (save_safetensors_adapter)

Apple MLX

API: mlx.core.save_safetensors

Strategy: Direct Mapping

Official Docs ↗

Flax NNX

API: —

Strategy: Macro 'save_file({arrays}, {file}, metadata={metadata})'

PaxML / Praxis

API: —

Strategy: Macro 'save_file({arrays}, {file}, metadata={metadata})'