search

ml-explore / mlx

MLX: An array framework for Apple silicon
https://ml-explore.github.io/mlx/
MIT License
14.83k stars 845 forks source link

Reset peak memory #1074

Closed awni closed 2 weeks ago

awni commented 2 weeks ago

Add a reset function to make memory profiling easier.