Files
mlx/benchmarks/python/batch_matmul_bench.py