Files
mlx/benchmarks/python/conv3d_train_bench_cpu.py