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
Flax NNX
API: —
Strategy: Macro 'save_file({arrays}, {file}, metadata={metadata})'
PaxML / Praxis
API: —
Strategy: Macro 'save_file({arrays}, {file}, metadata={metadata})'