How to time code in mlx in order to optimize performance and how to measure memory consumption?

[kyrollosyanny]

Hi all, I wanted to time different parts of my code to see where the bottleneck is. Given that mlx only executes when an eval is invoked, Im finding it hard trusting the timing numbers. I was wondering what the best practices for timing is.

In addition, I was wondering if there is a recommended way to measure memory consumption of code.

Thank you