GetPeakMemory

Get the peak amount of used memory in bytes.

PyTorch

API: torch.cuda.max_memory_allocated
Strategy: Direct Mapping
Official Docs ↗

Apple MLX

API: mlx.core.get_peak_memory
Strategy: Direct Mapping
Official Docs ↗