Files
mlx/benchmarks/python/rms_norm_bench.py