GetPeakMemory ============= Get the peak amount of used memory in bytes. .. raw:: html

PyTorch

API: torch.cuda.max_memory_allocated
Strategy: Direct Mapping

Apple MLX

API: mlx.core.get_peak_memory
Strategy: Direct Mapping