GetPeakMemory ============= Get the peak amount of used memory in bytes. .. raw:: html
torch.cuda.max_memory_allocated
mlx.core.get_peak_memory