jax

8c2901cf - Add `--export_dir` and `--baseline_dir` flags to benchmark.py. (#2677)

Add `--export_dir` and `--baseline_dir` flags to benchmark.py. (#2677) `--export_dir` allows saving benchmark results to CSV files, and `--baseline_dir` allows comparing results to a baseline exported via `--export_dir`.