TorchSave

Definition for TorchSave

Abstract Signature:

TorchSave(obj, f)

PyTorchJAX (Core)NumPy

PyTorch

API: torch.save

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: save

Strategy: Plugin (io_handler)

NumPy

API: save

Strategy: Plugin (io_handler)