Files
mlx-examples/esm/benchmarks/benchmark_pt.py