MemoryFormat

Enum for memory layout.

PyTorchJAX (Core)

PyTorch

API: torch.memory_format

Strategy: Direct Mapping

Official Docs ↗

JAX (Core)

API: —

Strategy: Custom / Partial