Function resetPeakMemory

mx.reset_peak_memory — zero the peak counter (e.g. after model load, so a subsequent peakMemory() reads generation-only peak).