MemoryFormat ============ Enum for memory layout. .. raw:: html

PyTorch

API: torch.memory_format
Strategy: Direct Mapping

JAX (Core)

API:
Strategy: Custom / Partial