Add --configs flag to run_and_save_all_benchmarks.py to allow selective runs. - #2494

Add --configs flag to run_and_save_all_benchmarks.py to allow selective runs.#2494

copybara-service[bot] merged 1 commit into

maingoogle-deepmind/torax:mainfrom

test_986811081google-deepmind/torax:test_986811081Copy head branch name to clipboard