├── .assets ├── mlc_logo.png └── performance_profiles.png ├── .dockerignore ├── .github ├── CODEOWNERS ├── ISSUE_TEMPLATE │ ├── bug_report.md │ ├── feature_request.md │ └── new_workload.md └── workflows │ ├── CI.yml │ ├── cla.yml │ ├── linting.yml │ ├── regression_tests.yml │ └── traindiffs_tests.yml ├── .gitignore ├── .pre-commit-config.yaml ├── LICENSE.md ├── README.md ├── __init__.py ├── algoperf ├── __init__.py ├── checkpoint_utils.py ├── data_utils.py ├── halton.py ├── init_utils.py ├── interop_utils.py ├── jax_sharding_utils.py ├── jax_utils.py ├── logger_utils.py ├── param_utils.py ├── profiler.py ├── pytorch_utils.py ├── random_utils.py ├── spec.py └── workloads │ ├── __init__.py │ ├── cifar │ ├── __init__.py │ ├── cifar_jax │ │ ├── __init__.py │ │ ├── input_pipeline.py │ │ ├── models.py │ │ └── workload.py │ ├── cifar_pytorch │ │ ├── __init__.py │ │ ├── models.py │ │ └── workload.py │ └── workload.py │ ├── criteo1tb │ ├── __init__.py │ ├── criteo1tb_jax │ │ ├── __init__.py │ │ ├── models.py │ │ └── workload.py │ ├── criteo1tb_pytorch │ │ ├── __init__.py │ │ ├── models.py │ │ └── workload.py │ ├── input_pipeline.py │ └── workload.py │ ├── fastmri │ ├── __init__.py │ ├── fastmri_jax │ │ ├── __init__.py │ │ ├── models.py │ │ ├── ssim.py │ │ └── workload.py │ ├── fastmri_pytorch │ │ ├── __init__.py │ │ ├── models.py │ │ ├── ssim.py │ │ └── workload.py │ ├── input_pipeline.py │ └── workload.py │ ├── imagenet_resnet │ ├── __init__.py │ ├── imagenet_jax │ │ ├── __init__.py │ │ ├── custom_tf_addons.py │ │ ├── input_pipeline.py │ │ ├── models.py │ │ ├── randaugment.py │ │ └── workload.py │ ├── imagenet_pytorch │ │ ├── __init__.py │ │ ├── models.py │ │ ├── randaugment.py │ │ └── workload.py │ ├── imagenet_v2.py │ └── workload.py │ ├── imagenet_vit │ ├── __init__.py │ ├── imagenet_jax │ │ ├── __init__.py │ │ ├── models.py │ │ └── workload.py │ ├── imagenet_pytorch │ │ ├── __init__.py │ │ ├── models.py │ │ └── workload.py │ └── workload.py │ ├── librispeech_conformer │ ├── __init__.py │ ├── input_pipeline.py │ ├── librispeech_jax │ │ ├── __init__.py │ │ ├── librispeech_preprocessor.py │ │ ├── models.py │ │ ├── spectrum_augmenter.py │ │ └── workload.py │ ├── librispeech_pytorch │ │ ├── __init__.py │ │ ├── models.py │ │ ├── preprocessor.py │ │ ├── spectrum_augmenter.py │ │ └── workload.py │ ├── metrics.py │ └── workload.py │ ├── librispeech_deepspeech │ ├── __init__.py │ ├── librispeech_jax │ │ ├── __init__.py │ │ ├── lstm.py │ │ ├── models.py │ │ └── workload.py │ └── librispeech_pytorch │ │ ├── __init__.py │ │ ├── models.py │ │ └── workload.py │ ├── mnist │ ├── __init__.py │ ├── mnist_jax │ │ ├── __init__.py │ │ └── workload.py │ ├── mnist_pytorch │ │ ├── __init__.py │ │ └── workload.py │ └── workload.py │ ├── ogbg │ ├── __init__.py │ ├── input_pipeline.py │ ├── metrics.py │ ├── ogbg_jax │ │ ├── __init__.py │ │ ├── models.py │ │ └── workload.py │ ├── ogbg_pytorch │ │ ├── __init__.py │ │ ├── models.py │ │ └── workload.py │ └── workload.py │ ├── utils.py │ ├── wmt │ ├── __init__.py │ ├── bleu.py │ ├── input_pipeline.py │ ├── tokenizer.py │ ├── wmt_jax │ │ ├── __init__.py │ │ ├── decode.py │ │ ├── models.py │ │ └── workload.py │ ├── wmt_pytorch │ │ ├── __init__.py │ │ ├── decode.py │ │ ├── models.py │ │ └── workload.py │ └── workload.py │ └── workloads.py ├── algorithms ├── README.md ├── __init__.py ├── archived_paper_baselines │ ├── README.md │ ├── __init__.py │ ├── adafactor │ │ ├── __init__.py │ │ ├── jax │ │ │ ├── __init__.py │ │ │ ├── sharded_adafactor.py │ │ │ └── submission.py │ │ ├── pytorch │ │ │ ├── __init__.py │ │ │ └── submission.py │ │ ├── tuning_search_space.json │ │ └── tuning_search_space_no_beta1.json │ ├── adamw │ │ ├── __init__.py │ │ ├── jax │ │ │ ├── __init__.py │ │ │ └── submission.py │ │ ├── pytorch │ │ │ ├── __init__.py │ │ │ └── submission.py │ │ ├── tuning_search_space.json │ │ └── tuning_search_space_no_beta1.json │ ├── lamb │ │ ├── __init__.py │ │ ├── jax │ │ │ ├── __init__.py │ │ │ └── submission.py │ │ ├── pytorch │ │ │ ├── __init__.py │ │ │ └── submission.py │ │ ├── tuning_search_space.json │ │ └── tuning_search_space_no_beta1.json │ ├── momentum │ │ ├── __init__.py │ │ ├── jax │ │ │ ├── __init__.py │ │ │ └── submission.py │ │ ├── pytorch │ │ │ ├── __init__.py │ │ │ └── submission.py │ │ ├── tuning_search_space.json │ │ └── tuning_search_space_no_beta1.json │ ├── nadamw │ │ ├── __init__.py │ │ ├── jax │ │ │ ├── __init__.py │ │ │ └── submission.py │ │ ├── pytorch │ │ │ ├── __init__.py │ │ │ └── submission.py │ │ ├── tuning_search_space.json │ │ └── tuning_search_space_no_beta1.json │ ├── nesterov │ │ ├── __init__.py │ │ ├── jax │ │ │ ├── __init__.py │ │ │ └── submission.py │ │ ├── pytorch │ │ │ ├── __init__.py │ │ │ └── submission.py │ │ ├── tuning_search_space.json │ │ └── tuning_search_space_no_beta1.json │ ├── sam │ │ ├── __init__.py │ │ ├── jax │ │ │ ├── __init__.py │ │ │ └── submission.py │ │ ├── pytorch │ │ │ ├── __init__.py │ │ │ └── submission.py │ │ ├── tuning_search_space.json │ │ └── tuning_search_space_no_beta1.json │ └── shampoo │ │ ├── __init__.py │ │ ├── jax │ │ ├── __init__.py │ │ ├── distributed_shampoo.py │ │ └── submission.py │ │ ├── pytorch │ │ └── __init__.py │ │ ├── tuning_search_space.json │ │ └── tuning_search_space_no_beta1.json ├── baselines │ ├── README.md │ ├── external_tuning │ │ ├── jax_nadamw_full_budget.py │ │ ├── pytorch_nadamw_full_budget.py │ │ └── tuning_search_space.json │ ├── logs │ │ ├── external_tuning │ │ │ └── baseline │ │ │ │ ├── study_0 │ │ │ │ ├── criteo1tb_jax │ │ │ │ │ ├── criteo1tb_jax_02-13-2024-13-02-24.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── fastmri_jax │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── imagenet_resnet_jax │ │ │ │ │ ├── imagenet_resnet_jax_01-26-2024-19-01-13.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── imagenet_vit_jax │ │ │ │ │ ├── imagenet_vit_jax_01-30-2024-13-40-46.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── librispeech_conformer_jax │ │ │ │ │ ├── librispeech_conformer_jax_02-14-2024-01-22-23.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── librispeech_deepspeech_jax │ │ │ │ │ ├── librispeech_deepspeech_jax_02-10-2024-22-49-44.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── ogbg_jax │ │ │ │ │ ├── ogbg_jax_02-04-2024-21-11-51.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ └── wmt_jax │ │ │ │ │ ├── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_5 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── wmt_jax_02-06-2024-10-57-53.log │ │ │ │ ├── study_1 │ │ │ │ ├── criteo1tb_jax │ │ │ │ │ ├── criteo1tb_jax_02-13-2024-14-28-38.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── fastmri_jax │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── imagenet_resnet_jax │ │ │ │ │ ├── imagenet_resnet_jax_01-26-2024-19-02-48.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── imagenet_vit_jax │ │ │ │ │ ├── imagenet_vit_jax_01-30-2024-13-42-02.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── librispeech_conformer_jax │ │ │ │ │ ├── librispeech_conformer_jax_02-14-2024-02-48-34.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── librispeech_deepspeech_jax │ │ │ │ │ ├── librispeech_deepspeech_jax_02-11-2024-00-20-46.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── ogbg_jax │ │ │ │ │ ├── ogbg_jax_02-04-2024-21-48-01.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ └── wmt_jax │ │ │ │ │ ├── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_5 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── wmt_jax_02-06-2024-11-39-02.log │ │ │ │ ├── study_2 │ │ │ │ ├── criteo1tb_jax │ │ │ │ │ ├── criteo1tb_jax_02-13-2024-14-24-50.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── fastmri_jax │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── imagenet_resnet_jax │ │ │ │ │ ├── imagenet_resnet_jax_01-26-2024-19-02-57.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── imagenet_vit_jax │ │ │ │ │ ├── imagenet_vit_jax_01-30-2024-13-48-02.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── librispeech_conformer_jax │ │ │ │ │ ├── librispeech_conformer_jax_02-14-2024-02-44-53.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── librispeech_deepspeech_jax │ │ │ │ │ ├── librispeech_deepspeech_jax_02-11-2024-00-46-57.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── ogbg_jax │ │ │ │ │ ├── ogbg_jax_02-04-2024-23-24-16.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ └── wmt_jax │ │ │ │ │ ├── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_5 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── wmt_jax_02-06-2024-13-15-17.log │ │ │ │ ├── study_3 │ │ │ │ ├── criteo1tb_jax │ │ │ │ │ ├── criteo1tb_jax_02-17-2024-11-56-37.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── fastmri_jax │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── imagenet_resnet_jax │ │ │ │ │ ├── imagenet_resnet_jax_01-26-2024-19-03-07.log │ │ │ │ │ ├── imagenet_resnet_jax_01-30-2024-20-48-52.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── imagenet_vit_jax │ │ │ │ │ ├── imagenet_vit_jax_01-26-2024-19-08-22.log │ │ │ │ │ ├── imagenet_vit_jax_02-03-2024-15-24-33.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── librispeech_conformer_jax │ │ │ │ │ ├── librispeech_conformer_jax_02-18-2024-00-16-19.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── librispeech_deepspeech_jax │ │ │ │ │ ├── librispeech_deepspeech_jax_02-15-2024-02-44-09.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── ogbg_jax │ │ │ │ │ ├── ogbg_jax_02-09-2024-01-06-01.log │ │ │ │ │ ├── trial_1 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── trial_5 │ │ │ │ │ │ ├── flags_0.json │ │ │ │ │ │ ├── hparams.json │ │ │ │ │ │ └── meta_data_0.json │ │ │ │ └── wmt_jax │ │ │ │ │ ├── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_2 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_3 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_4 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ ├── trial_5 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ │ └── wmt_jax_02-10-2024-14-47-06.log │ │ │ │ └── study_4 │ │ │ │ ├── criteo1tb_jax │ │ │ │ ├── criteo1tb_jax_02-13-2024-14-38-52.log │ │ │ │ ├── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_2 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_3 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_4 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ └── trial_5 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── fastmri_jax │ │ │ │ ├── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_2 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_3 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_4 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ └── trial_5 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── imagenet_resnet_jax │ │ │ │ ├── imagenet_resnet_jax_01-27-2024-01-35-38.log │ │ │ │ ├── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_2 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_3 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_4 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ └── trial_5 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── imagenet_vit_jax │ │ │ │ ├── imagenet_vit_jax_01-30-2024-19-56-30.log │ │ │ │ ├── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_2 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_3 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_4 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ └── trial_5 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── librispeech_conformer_jax │ │ │ │ ├── librispeech_conformer_jax_02-14-2024-02-58-52.log │ │ │ │ ├── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_2 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_3 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_4 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ └── trial_5 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── librispeech_deepspeech_jax │ │ │ │ ├── librispeech_deepspeech_jax_02-11-2024-04-05-57.log │ │ │ │ ├── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_2 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_3 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_4 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ └── trial_5 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── ogbg_jax │ │ │ │ ├── ogbg_jax_02-05-2024-03-48-25.log │ │ │ │ ├── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_2 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_3 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_4 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ └── trial_5 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ ├── hparams.json │ │ │ │ │ └── meta_data_0.json │ │ │ │ └── wmt_jax │ │ │ │ ├── trial_1 │ │ │ │ ├── flags_0.json │ │ │ │ ├── hparams.json │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_2 │ │ │ │ ├── flags_0.json │ │ │ │ ├── hparams.json │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_3 │ │ │ │ ├── flags_0.json │ │ │ │ ├── hparams.json │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_4 │ │ │ │ ├── flags_0.json │ │ │ │ ├── hparams.json │ │ │ │ └── meta_data_0.json │ │ │ │ ├── trial_5 │ │ │ │ ├── flags_0.json │ │ │ │ ├── hparams.json │ │ │ │ └── meta_data_0.json │ │ │ │ └── wmt_jax_02-06-2024-17-34-24.log │ │ └── self_tuning │ │ │ └── baseline │ │ │ ├── study_0 │ │ │ ├── criteo1tb_jax │ │ │ │ ├── criteo1tb_jax_03-13-2024-10-14-45.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── fastmri_jax │ │ │ │ ├── fastmri_jax_03-12-2024-03-47-57.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── imagenet_resnet_jax │ │ │ │ ├── imagenet_resnet_jax_02-29-2024-05-10-31.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── imagenet_vit_jax │ │ │ │ ├── imagenet_vit_jax_03-02-2024-11-32-59.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── librispeech_conformer_jax │ │ │ │ ├── librispeech_conformer_jax_03-13-2024-20-09-52.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── librispeech_deepspeech_jax │ │ │ │ ├── librispeech_deepspeech_jax_03-12-2024-23-07-50.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── ogbg_jax │ │ │ │ ├── ogbg_jax_03-05-2024-10-01-36.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ └── wmt_jax │ │ │ │ ├── trial_1 │ │ │ │ ├── flags_0.json │ │ │ │ └── meta_data_0.json │ │ │ │ └── wmt_jax_03-12-2024-04-28-16.log │ │ │ ├── study_1 │ │ │ ├── criteo1tb_jax │ │ │ │ ├── criteo1tb_jax_03-13-2024-10-39-31.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── fastmri_jax │ │ │ │ ├── fastmri_jax_03-12-2024-03-32-44.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── imagenet_resnet_jax │ │ │ │ ├── imagenet_resnet_jax_02-29-2024-05-32-47.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── imagenet_vit_jax │ │ │ │ ├── imagenet_vit_jax_03-02-2024-11-55-17.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── librispeech_conformer_jax │ │ │ │ ├── librispeech_conformer_jax_03-13-2024-19-59-18.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── librispeech_deepspeech_jax │ │ │ │ ├── librispeech_deepspeech_jax_03-12-2024-23-37-44.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── ogbg_jax │ │ │ │ ├── ogbg_jax_03-05-2024-10-18-35.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ └── wmt_jax │ │ │ │ ├── trial_1 │ │ │ │ ├── flags_0.json │ │ │ │ └── meta_data_0.json │ │ │ │ └── wmt_jax_03-12-2024-04-28-04.log │ │ │ ├── study_2 │ │ │ ├── criteo1tb_jax │ │ │ │ ├── criteo1tb_jax_03-13-2024-09-51-13.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── fastmri_jax │ │ │ │ ├── fastmri_jax_03-12-2024-03-34-27.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── imagenet_resnet_jax │ │ │ │ ├── imagenet_resnet_jax_02-29-2024-05-13-39.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── imagenet_vit_jax │ │ │ │ ├── imagenet_vit_jax_03-02-2024-11-06-03.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── librispeech_conformer_jax │ │ │ │ ├── librispeech_conformer_jax_03-13-2024-18-55-53.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── librispeech_deepspeech_jax │ │ │ │ ├── librispeech_deepspeech_jax_03-12-2024-22-49-22.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── ogbg_jax │ │ │ │ ├── ogbg_jax_03-05-2024-10-04-38.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ └── wmt_jax │ │ │ │ ├── trial_1 │ │ │ │ ├── flags_0.json │ │ │ │ └── meta_data_0.json │ │ │ │ └── wmt_jax_03-12-2024-04-14-45.log │ │ │ ├── study_3 │ │ │ ├── criteo1tb_jax │ │ │ │ ├── criteo1tb_jax_03-16-2024-12-03-01.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── fastmri_jax │ │ │ │ ├── fastmri_jax_03-12-2024-03-34-36.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── imagenet_resnet_jax │ │ │ │ ├── imagenet_resnet_jax_02-29-2024-05-13-57.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── imagenet_vit_jax │ │ │ │ ├── imagenet_vit_jax_03-02-2024-11-31-55.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── librispeech_conformer_jax │ │ │ │ ├── librispeech_conformer_jax_03-16-2024-21-22-50.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── librispeech_deepspeech_jax │ │ │ │ ├── librispeech_deepspeech_jax_03-15-2024-00-20-37.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ ├── ogbg_jax │ │ │ │ ├── ogbg_jax_03-05-2024-10-10-44.log │ │ │ │ └── trial_1 │ │ │ │ │ ├── flags_0.json │ │ │ │ │ └── meta_data_0.json │ │ │ └── wmt_jax │ │ │ │ ├── trial_1 │ │ │ │ ├── flags_0.json │ │ │ │ └── meta_data_0.json │ │ │ │ └── wmt_jax_03-12-2024-04-09-55.log │ │ │ └── study_4 │ │ │ ├── criteo1tb_jax │ │ │ ├── criteo1tb_jax_03-13-2024-12-00-36.log │ │ │ └── trial_1 │ │ │ │ ├── flags_0.json │ │ │ │ └── meta_data_0.json │ │ │ ├── fastmri_jax │ │ │ ├── fastmri_jax_03-12-2024-03-27-32.log │ │ │ └── trial_1 │ │ │ │ ├── flags_0.json │ │ │ │ └── meta_data_0.json │ │ │ ├── imagenet_resnet_jax │ │ │ ├── imagenet_resnet_jax_03-05-2024-22-45-59.log │ │ │ └── trial_1 │ │ │ │ ├── flags_0.json │ │ │ │ └── meta_data_0.json │ │ │ ├── imagenet_vit_jax │ │ │ ├── imagenet_vit_jax_03-08-2024-05-00-35.log │ │ │ └── trial_1 │ │ │ │ ├── flags_0.json │ │ │ │ └── meta_data_0.json │ │ │ ├── librispeech_conformer_jax │ │ │ ├── librispeech_conformer_jax_03-13-2024-20-34-38.log │ │ │ └── trial_1 │ │ │ │ ├── flags_0.json │ │ │ │ └── meta_data_0.json │ │ │ ├── librispeech_deepspeech_jax │ │ │ ├── librispeech_deepspeech_jax_03-13-2024-00-19-52.log │ │ │ └── trial_1 │ │ │ │ ├── flags_0.json │ │ │ │ └── meta_data_0.json │ │ │ ├── ogbg_jax │ │ │ ├── ogbg_jax_03-11-2024-03-14-14.log │ │ │ └── trial_1 │ │ │ │ ├── flags_0.json │ │ │ │ └── meta_data_0.json │ │ │ └── wmt_jax │ │ │ ├── trial_1 │ │ │ ├── flags_0.json │ │ │ └── meta_data_0.json │ │ │ └── wmt_jax_03-12-2024-04-02-58.log │ └── self_tuning │ │ ├── jax_nadamw_full_budget.py │ │ └── pytorch_nadamw_full_budget.py ├── development_algorithms │ ├── __init__.py │ ├── cifar │ │ ├── __init__.py │ │ ├── cifar_jax │ │ │ ├── __init__.py │ │ │ └── submission.py │ │ ├── cifar_pytorch │ │ │ ├── __init__.py │ │ │ └── submission.py │ │ └── tuning_search_space.json │ └── mnist │ │ ├── __init__.py │ │ ├── discrete_space.json │ │ ├── mnist_jax │ │ ├── __init__.py │ │ └── submission.py │ │ ├── mnist_pytorch │ │ ├── __init__.py │ │ └── submission.py │ │ └── tuning_search_space.json ├── submission_checker.py ├── target_setting_algorithms │ ├── README.md │ ├── __init__.py │ ├── cosine_warmup.py │ ├── criteo1tb │ │ └── tuning_search_space.json │ ├── criteo1tb_embed_init │ │ └── tuning_search_space.json │ ├── criteo1tb_layernorm │ │ └── tuning_search_space.json │ ├── criteo1tb_resnet │ │ └── tuning_search_space.json │ ├── data_selection.py │ ├── fastmri │ │ └── tuning_search_space.json │ ├── fastmri_layernorm │ │ └── tuning_search_space.json │ ├── fastmri_model_size │ │ └── tuning_search_space.json │ ├── fastmri_tanh │ │ └── tuning_search_space.json │ ├── get_batch_size.py │ ├── imagenet_resnet │ │ └── tuning_search_space.json │ ├── imagenet_resnet_gelu │ │ └── tuning_search_space.json │ ├── imagenet_resnet_large_bn_init │ │ └── tuning_search_space.json │ ├── imagenet_resnet_silu │ │ └── tuning_search_space.json │ ├── imagenet_vit │ │ └── tuning_search_space.json │ ├── imagenet_vit_glu │ │ └── tuning_search_space.json │ ├── imagenet_vit_map │ │ └── tuning_search_space.json │ ├── imagenet_vit_post_ln │ │ └── tuning_search_space.json │ ├── jax_adamw.py │ ├── jax_momentum.py │ ├── jax_nadamw.py │ ├── jax_nesterov.py │ ├── jax_submission_base.py │ ├── librispeech_conformer │ │ └── tuning_search_space.json │ ├── librispeech_conformer_attention_temperature │ │ └── tuning_search_space.json │ ├── librispeech_conformer_gelu │ │ └── tuning_search_space.json │ ├── librispeech_conformer_layernorm │ │ └── tuning_search_space.json │ ├── librispeech_deepspeech │ │ ├── librispeech_deepspeech_no_resnet │ │ │ └── tuning_search_space.json │ │ ├── librispeech_deepspeech_norm_and_spec_aug │ │ │ └── tuning_search_space.json │ │ └── tuning_search_space.json │ ├── librispeech_deepspeech_no_resnet │ │ └── tuning_search_space.json │ ├── librispeech_deepspeech_norm_and_spec_aug │ │ └── tuning_search_space.json │ ├── librispeech_deepspeech_tanh │ │ └── tuning_search_space.json │ ├── ogbg │ │ └── tuning_search_space.json │ ├── ogbg_gelu │ │ └── tuning_search_space.json │ ├── ogbg_model_size │ │ └── tuning_search_space.json │ ├── ogbg_silu │ │ └── tuning_search_space.json │ ├── pytorch_adamw.py │ ├── pytorch_momentum.py │ ├── pytorch_nadamw.py │ ├── pytorch_nesterov.py │ ├── pytorch_submission_base.py │ ├── wmt │ │ └── tuning_search_space.json │ ├── wmt_attention_temp │ │ └── tuning_search_space.json │ ├── wmt_glu_tanh │ │ └── tuning_search_space.json │ └── wmt_post_ln │ │ └── tuning_search_space.json └── template │ └── submission.py ├── datasets ├── README.md ├── dataset_setup.py ├── librispeech_preprocess.py └── librispeech_tokenizer.py ├── docker ├── Dockerfile ├── Singularity.def ├── build_docker_images.sh └── scripts │ ├── check_gpu.py │ ├── cloud-init.cfg │ ├── cloud-startup.sh │ ├── singularity_converter.py │ ├── startup.sh │ └── test_startup.sh ├── docs ├── CHANGELOG.md ├── CONTRIBUTING.md ├── DOCUMENTATION.md └── GETTING_STARTED.md ├── pyproject.toml ├── scoring ├── __init__.py ├── algoperf_v05 │ ├── generate_held_out_workloads.py │ ├── held_out_workloads_algoperf_v05.json │ ├── held_out_workloads_example.json │ └── score_submissions.py ├── compute_speedups.py ├── performance_profile.py ├── score_submissions.py ├── scoring_utils.py ├── test_data │ ├── adamw_fastmri_jax_04-18-2023-13-10-58.log │ └── experiment_dir │ │ └── study_0 │ │ └── mnist_jax │ │ ├── trial_0 │ │ └── eval_measurements.csv │ │ └── trial_1 │ │ └── eval_measurements.csv ├── test_scoring_utils.py └── utils │ ├── export_runs_to_wandb.py │ ├── package_logs.py │ ├── run_workloads.py │ ├── slurm │ ├── README.md │ ├── algoperf_slurm_cluster.yaml │ ├── algoperf_slurm_pakcer_builder.yaml │ ├── config.json │ ├── make_job_config.py │ └── run_jobs.sh │ ├── workload_metadata_external_tuning.json │ └── workload_metadata_self_tuning.json ├── submission_runner.py └── tests ├── __init__.py ├── modeldiffs ├── __init__.py ├── criteo1tb │ ├── __init__.py │ └── compare.py ├── criteo1tb_embed_init │ ├── __init__.py │ └── compare.py ├── criteo1tb_layernorm │ ├── __init__.py │ └── compare.py ├── criteo1tb_resnet │ ├── __init__.py │ └── compare.py ├── diff.py ├── fastmri │ ├── __init__.py │ └── compare.py ├── fastmri_layernorm │ ├── __init__.py │ └── compare.py ├── fastmri_model_size │ ├── __init__.py │ └── compare.py ├── fastmri_tanh │ ├── __init__.py │ └── compare.py ├── imagenet_resnet │ ├── __init__.py │ ├── compare.py │ ├── gelu_compare.py │ └── silu_compare.py ├── imagenet_vit │ ├── __init__.py │ └── compare.py ├── imagenet_vit_glu │ ├── __init__.py │ └── compare.py ├── imagenet_vit_map │ ├── __init__.py │ └── compare.py ├── imagenet_vit_postln │ ├── __init__.py │ └── compare.py ├── librispeech_conformer │ ├── __init__.py │ └── compare.py ├── librispeech_conformer_attention_temperature │ ├── __init__.py │ └── compare.py ├── librispeech_conformer_gelu │ ├── __init__.py │ └── compare.py ├── librispeech_conformer_layernorm │ ├── __init__.py │ └── compare.py ├── librispeech_deepspeech │ ├── __init__.py │ └── compare.py ├── librispeech_deepspeech_noresnet │ ├── __init__.py │ └── compare.py ├── librispeech_deepspeech_normaug │ ├── __init__.py │ └── compare.py ├── librispeech_deepspeech_tanh │ ├── __init__.py │ └── compare.py ├── ogbg │ ├── __init__.py │ └── compare.py ├── ogbg_gelu │ ├── __init__.py │ └── compare.py ├── ogbg_model_size │ ├── __init__.py │ └── compare.py ├── ogbg_silu │ ├── __init__.py │ └── compare.py ├── torch2jax_utils.py ├── vanilla_sgd_jax.py ├── vanilla_sgd_pytorch.py ├── wmt │ ├── __init__.py │ └── compare.py ├── wmt_attention_temp │ ├── __init__.py │ └── compare.py ├── wmt_glu_tanh │ ├── __init__.py │ └── compare.py └── wmt_post_ln │ ├── __init__.py │ └── compare.py ├── reference_algorithm_tests.py ├── submission_runner_test.py ├── test_baselines.py ├── test_jax_utils.py ├── test_num_params.py ├── test_param_shapes.py ├── test_param_types.py ├── test_ssim.py ├── test_traindiffs.py ├── test_version.py └── workloads ├── __init__.py └── imagenet_resnet ├── __init__.py └── imagenet_jax ├── __init__.py └── workload_test.py /.assets/mlc_logo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/.assets/mlc_logo.png -------------------------------------------------------------------------------- /.assets/performance_profiles.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/.assets/performance_profiles.png -------------------------------------------------------------------------------- /.dockerignore: -------------------------------------------------------------------------------- 1 | * 2 | !datasets/ 3 | !docker/ -------------------------------------------------------------------------------- /.github/CODEOWNERS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/.github/CODEOWNERS -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/bug_report.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/.github/ISSUE_TEMPLATE/bug_report.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/feature_request.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/.github/ISSUE_TEMPLATE/feature_request.md -------------------------------------------------------------------------------- /.github/ISSUE_TEMPLATE/new_workload.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/.github/ISSUE_TEMPLATE/new_workload.md -------------------------------------------------------------------------------- /.github/workflows/CI.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/.github/workflows/CI.yml -------------------------------------------------------------------------------- /.github/workflows/cla.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/.github/workflows/cla.yml -------------------------------------------------------------------------------- /.github/workflows/linting.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/.github/workflows/linting.yml -------------------------------------------------------------------------------- /.github/workflows/regression_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/.github/workflows/regression_tests.yml -------------------------------------------------------------------------------- /.github/workflows/traindiffs_tests.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/.github/workflows/traindiffs_tests.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/__init__.py -------------------------------------------------------------------------------- /algoperf/checkpoint_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/checkpoint_utils.py -------------------------------------------------------------------------------- /algoperf/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/data_utils.py -------------------------------------------------------------------------------- /algoperf/halton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/halton.py -------------------------------------------------------------------------------- /algoperf/init_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/init_utils.py -------------------------------------------------------------------------------- /algoperf/interop_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/interop_utils.py -------------------------------------------------------------------------------- /algoperf/jax_sharding_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/jax_sharding_utils.py -------------------------------------------------------------------------------- /algoperf/jax_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/jax_utils.py -------------------------------------------------------------------------------- /algoperf/logger_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/logger_utils.py -------------------------------------------------------------------------------- /algoperf/param_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/param_utils.py -------------------------------------------------------------------------------- /algoperf/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/profiler.py -------------------------------------------------------------------------------- /algoperf/pytorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/pytorch_utils.py -------------------------------------------------------------------------------- /algoperf/random_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/random_utils.py -------------------------------------------------------------------------------- /algoperf/spec.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/spec.py -------------------------------------------------------------------------------- /algoperf/workloads/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/cifar/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/cifar/cifar_jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/cifar/cifar_jax/input_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/cifar/cifar_jax/input_pipeline.py -------------------------------------------------------------------------------- /algoperf/workloads/cifar/cifar_jax/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/cifar/cifar_jax/models.py -------------------------------------------------------------------------------- /algoperf/workloads/cifar/cifar_jax/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/cifar/cifar_jax/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/cifar/cifar_pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/cifar/cifar_pytorch/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/cifar/cifar_pytorch/models.py -------------------------------------------------------------------------------- /algoperf/workloads/cifar/cifar_pytorch/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/cifar/cifar_pytorch/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/cifar/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/cifar/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/criteo1tb/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/criteo1tb/criteo1tb_jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/criteo1tb/criteo1tb_jax/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/criteo1tb/criteo1tb_jax/models.py -------------------------------------------------------------------------------- /algoperf/workloads/criteo1tb/criteo1tb_jax/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/criteo1tb/criteo1tb_jax/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/criteo1tb/criteo1tb_pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/criteo1tb/criteo1tb_pytorch/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/criteo1tb/criteo1tb_pytorch/models.py -------------------------------------------------------------------------------- /algoperf/workloads/criteo1tb/criteo1tb_pytorch/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/criteo1tb/criteo1tb_pytorch/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/criteo1tb/input_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/criteo1tb/input_pipeline.py -------------------------------------------------------------------------------- /algoperf/workloads/criteo1tb/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/criteo1tb/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/fastmri/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/fastmri/fastmri_jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/fastmri/fastmri_jax/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/fastmri/fastmri_jax/models.py -------------------------------------------------------------------------------- /algoperf/workloads/fastmri/fastmri_jax/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/fastmri/fastmri_jax/ssim.py -------------------------------------------------------------------------------- /algoperf/workloads/fastmri/fastmri_jax/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/fastmri/fastmri_jax/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/fastmri/fastmri_pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/fastmri/fastmri_pytorch/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/fastmri/fastmri_pytorch/models.py -------------------------------------------------------------------------------- /algoperf/workloads/fastmri/fastmri_pytorch/ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/fastmri/fastmri_pytorch/ssim.py -------------------------------------------------------------------------------- /algoperf/workloads/fastmri/fastmri_pytorch/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/fastmri/fastmri_pytorch/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/fastmri/input_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/fastmri/input_pipeline.py -------------------------------------------------------------------------------- /algoperf/workloads/fastmri/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/fastmri/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_resnet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_resnet/imagenet_jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_resnet/imagenet_jax/custom_tf_addons.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/imagenet_resnet/imagenet_jax/custom_tf_addons.py -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_resnet/imagenet_jax/input_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/imagenet_resnet/imagenet_jax/input_pipeline.py -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_resnet/imagenet_jax/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/imagenet_resnet/imagenet_jax/models.py -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_resnet/imagenet_jax/randaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/imagenet_resnet/imagenet_jax/randaugment.py -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_resnet/imagenet_jax/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/imagenet_resnet/imagenet_jax/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_resnet/imagenet_pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_resnet/imagenet_pytorch/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/imagenet_resnet/imagenet_pytorch/models.py -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_resnet/imagenet_pytorch/randaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/imagenet_resnet/imagenet_pytorch/randaugment.py -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_resnet/imagenet_pytorch/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/imagenet_resnet/imagenet_pytorch/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_resnet/imagenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/imagenet_resnet/imagenet_v2.py -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_resnet/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/imagenet_resnet/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_vit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_vit/imagenet_jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_vit/imagenet_jax/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/imagenet_vit/imagenet_jax/models.py -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_vit/imagenet_jax/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/imagenet_vit/imagenet_jax/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_vit/imagenet_pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_vit/imagenet_pytorch/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/imagenet_vit/imagenet_pytorch/models.py -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_vit/imagenet_pytorch/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/imagenet_vit/imagenet_pytorch/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/imagenet_vit/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/imagenet_vit/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_conformer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_conformer/input_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/librispeech_conformer/input_pipeline.py -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_conformer/librispeech_jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_conformer/librispeech_jax/librispeech_preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/librispeech_conformer/librispeech_jax/librispeech_preprocessor.py -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_conformer/librispeech_jax/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/librispeech_conformer/librispeech_jax/models.py -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_conformer/librispeech_jax/spectrum_augmenter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/librispeech_conformer/librispeech_jax/spectrum_augmenter.py -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_conformer/librispeech_jax/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/librispeech_conformer/librispeech_jax/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_conformer/librispeech_pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_conformer/librispeech_pytorch/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/librispeech_conformer/librispeech_pytorch/models.py -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_conformer/librispeech_pytorch/preprocessor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/librispeech_conformer/librispeech_pytorch/preprocessor.py -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_conformer/librispeech_pytorch/spectrum_augmenter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/librispeech_conformer/librispeech_pytorch/spectrum_augmenter.py -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_conformer/librispeech_pytorch/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/librispeech_conformer/librispeech_pytorch/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_conformer/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/librispeech_conformer/metrics.py -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_conformer/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/librispeech_conformer/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_deepspeech/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_deepspeech/librispeech_jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_deepspeech/librispeech_jax/lstm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/librispeech_deepspeech/librispeech_jax/lstm.py -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_deepspeech/librispeech_jax/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/librispeech_deepspeech/librispeech_jax/models.py -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_deepspeech/librispeech_jax/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/librispeech_deepspeech/librispeech_jax/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_deepspeech/librispeech_pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_deepspeech/librispeech_pytorch/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/librispeech_deepspeech/librispeech_pytorch/models.py -------------------------------------------------------------------------------- /algoperf/workloads/librispeech_deepspeech/librispeech_pytorch/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/librispeech_deepspeech/librispeech_pytorch/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/mnist/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/mnist/mnist_jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/mnist/mnist_jax/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/mnist/mnist_jax/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/mnist/mnist_pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/mnist/mnist_pytorch/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/mnist/mnist_pytorch/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/mnist/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/mnist/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/ogbg/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/ogbg/input_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/ogbg/input_pipeline.py -------------------------------------------------------------------------------- /algoperf/workloads/ogbg/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/ogbg/metrics.py -------------------------------------------------------------------------------- /algoperf/workloads/ogbg/ogbg_jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/ogbg/ogbg_jax/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/ogbg/ogbg_jax/models.py -------------------------------------------------------------------------------- /algoperf/workloads/ogbg/ogbg_jax/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/ogbg/ogbg_jax/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/ogbg/ogbg_pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/ogbg/ogbg_pytorch/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/ogbg/ogbg_pytorch/models.py -------------------------------------------------------------------------------- /algoperf/workloads/ogbg/ogbg_pytorch/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/ogbg/ogbg_pytorch/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/ogbg/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/ogbg/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/utils.py -------------------------------------------------------------------------------- /algoperf/workloads/wmt/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/wmt/bleu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/wmt/bleu.py -------------------------------------------------------------------------------- /algoperf/workloads/wmt/input_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/wmt/input_pipeline.py -------------------------------------------------------------------------------- /algoperf/workloads/wmt/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/wmt/tokenizer.py -------------------------------------------------------------------------------- /algoperf/workloads/wmt/wmt_jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/wmt/wmt_jax/decode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/wmt/wmt_jax/decode.py -------------------------------------------------------------------------------- /algoperf/workloads/wmt/wmt_jax/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/wmt/wmt_jax/models.py -------------------------------------------------------------------------------- /algoperf/workloads/wmt/wmt_jax/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/wmt/wmt_jax/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/wmt/wmt_pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algoperf/workloads/wmt/wmt_pytorch/decode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/wmt/wmt_pytorch/decode.py -------------------------------------------------------------------------------- /algoperf/workloads/wmt/wmt_pytorch/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/wmt/wmt_pytorch/models.py -------------------------------------------------------------------------------- /algoperf/workloads/wmt/wmt_pytorch/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/wmt/wmt_pytorch/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/wmt/workload.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/wmt/workload.py -------------------------------------------------------------------------------- /algoperf/workloads/workloads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algoperf/workloads/workloads.py -------------------------------------------------------------------------------- /algorithms/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/README.md -------------------------------------------------------------------------------- /algorithms/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/README.md -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/adafactor/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/adafactor/jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/adafactor/jax/sharded_adafactor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/adafactor/jax/sharded_adafactor.py -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/adafactor/jax/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/adafactor/jax/submission.py -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/adafactor/pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/adafactor/pytorch/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/adafactor/pytorch/submission.py -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/adafactor/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/adafactor/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/adafactor/tuning_search_space_no_beta1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/adafactor/tuning_search_space_no_beta1.json -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/adamw/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/adamw/jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/adamw/jax/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/adamw/jax/submission.py -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/adamw/pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/adamw/pytorch/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/adamw/pytorch/submission.py -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/adamw/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/adamw/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/adamw/tuning_search_space_no_beta1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/adamw/tuning_search_space_no_beta1.json -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/lamb/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/lamb/jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/lamb/jax/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/lamb/jax/submission.py -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/lamb/pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/lamb/pytorch/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/lamb/pytorch/submission.py -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/lamb/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/lamb/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/lamb/tuning_search_space_no_beta1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/lamb/tuning_search_space_no_beta1.json -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/momentum/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/momentum/jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/momentum/jax/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/momentum/jax/submission.py -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/momentum/pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/momentum/pytorch/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/momentum/pytorch/submission.py -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/momentum/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/momentum/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/momentum/tuning_search_space_no_beta1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/momentum/tuning_search_space_no_beta1.json -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/nadamw/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/nadamw/jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/nadamw/jax/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/nadamw/jax/submission.py -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/nadamw/pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/nadamw/pytorch/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/nadamw/pytorch/submission.py -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/nadamw/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/nadamw/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/nadamw/tuning_search_space_no_beta1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/nadamw/tuning_search_space_no_beta1.json -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/nesterov/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/nesterov/jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/nesterov/jax/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/nesterov/jax/submission.py -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/nesterov/pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/nesterov/pytorch/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/nesterov/pytorch/submission.py -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/nesterov/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/nesterov/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/nesterov/tuning_search_space_no_beta1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/nesterov/tuning_search_space_no_beta1.json -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/sam/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/sam/jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/sam/jax/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/sam/jax/submission.py -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/sam/pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/sam/pytorch/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/sam/pytorch/submission.py -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/sam/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/sam/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/sam/tuning_search_space_no_beta1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/sam/tuning_search_space_no_beta1.json -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/shampoo/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/shampoo/jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/shampoo/jax/distributed_shampoo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/shampoo/jax/distributed_shampoo.py -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/shampoo/jax/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/shampoo/jax/submission.py -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/shampoo/pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/shampoo/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/shampoo/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/archived_paper_baselines/shampoo/tuning_search_space_no_beta1.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/archived_paper_baselines/shampoo/tuning_search_space_no_beta1.json -------------------------------------------------------------------------------- /algorithms/baselines/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/README.md -------------------------------------------------------------------------------- /algorithms/baselines/external_tuning/jax_nadamw_full_budget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/external_tuning/jax_nadamw_full_budget.py -------------------------------------------------------------------------------- /algorithms/baselines/external_tuning/pytorch_nadamw_full_budget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/external_tuning/pytorch_nadamw_full_budget.py -------------------------------------------------------------------------------- /algorithms/baselines/external_tuning/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/external_tuning/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/criteo1tb_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_2/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_2/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_3/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_3/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_4/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_4/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_5/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/fastmri_jax/trial_5/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_2/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_2/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_3/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_3/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_4/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_4/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_5/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/ogbg_jax/trial_5/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_2/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_2/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_3/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_3/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_4/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_4/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_5/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_0/wmt_jax/trial_5/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/criteo1tb_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_2/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_2/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_3/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_3/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_4/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_4/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_5/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/fastmri_jax/trial_5/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_2/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_2/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_3/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_3/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_4/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_4/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_5/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/ogbg_jax/trial_5/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_2/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_2/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_3/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_3/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_4/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_4/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_5/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_1/wmt_jax/trial_5/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/criteo1tb_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_2/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_2/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_3/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_3/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_4/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_4/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_5/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/fastmri_jax/trial_5/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_2/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_2/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_3/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_3/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_4/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_4/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_5/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/ogbg_jax/trial_5/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_2/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_2/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_3/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_3/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_4/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_4/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_5/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_2/wmt_jax/trial_5/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/criteo1tb_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_2/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_2/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_3/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_3/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_4/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_4/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_5/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/fastmri_jax/trial_5/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_2/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_2/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_3/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_3/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_4/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_4/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_5/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/ogbg_jax/trial_5/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_2/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_2/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_3/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_3/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_4/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_4/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_5/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_3/wmt_jax/trial_5/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/criteo1tb_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/fastmri_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_2/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_2/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_3/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_3/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_4/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_4/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_5/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/ogbg_jax/trial_5/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_1/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_1/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_2/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_2/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_2/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_2/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_2/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_2/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_3/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_3/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_3/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_3/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_3/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_3/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_4/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_4/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_4/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_4/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_4/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_4/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_5/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_5/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_5/hparams.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_5/hparams.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_5/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/external_tuning/baseline/study_4/wmt_jax/trial_5/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_0/criteo1tb_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_0/criteo1tb_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_0/fastmri_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_0/fastmri_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_0/fastmri_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_0/fastmri_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_0/imagenet_vit_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_0/imagenet_vit_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_0/ogbg_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_0/ogbg_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_0/ogbg_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_0/ogbg_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_0/wmt_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_0/wmt_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_0/wmt_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_0/wmt_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_1/criteo1tb_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_1/criteo1tb_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_1/fastmri_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_1/fastmri_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_1/fastmri_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_1/fastmri_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_1/imagenet_vit_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_1/imagenet_vit_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_1/ogbg_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_1/ogbg_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_1/ogbg_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_1/ogbg_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_1/wmt_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_1/wmt_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_1/wmt_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_1/wmt_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_2/criteo1tb_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_2/criteo1tb_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_2/fastmri_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_2/fastmri_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_2/fastmri_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_2/fastmri_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_2/imagenet_vit_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_2/imagenet_vit_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_2/ogbg_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_2/ogbg_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_2/ogbg_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_2/ogbg_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_2/wmt_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_2/wmt_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_2/wmt_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_2/wmt_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_3/criteo1tb_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_3/criteo1tb_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_3/fastmri_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_3/fastmri_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_3/fastmri_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_3/fastmri_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_3/imagenet_vit_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_3/imagenet_vit_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_3/ogbg_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_3/ogbg_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_3/ogbg_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_3/ogbg_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_3/wmt_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_3/wmt_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_3/wmt_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_3/wmt_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_4/criteo1tb_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_4/criteo1tb_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_4/fastmri_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_4/fastmri_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_4/fastmri_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_4/fastmri_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_4/imagenet_vit_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_4/imagenet_vit_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_4/ogbg_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_4/ogbg_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_4/ogbg_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_4/ogbg_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_4/wmt_jax/trial_1/flags_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_4/wmt_jax/trial_1/flags_0.json -------------------------------------------------------------------------------- /algorithms/baselines/logs/self_tuning/baseline/study_4/wmt_jax/trial_1/meta_data_0.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/logs/self_tuning/baseline/study_4/wmt_jax/trial_1/meta_data_0.json -------------------------------------------------------------------------------- /algorithms/baselines/self_tuning/jax_nadamw_full_budget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/self_tuning/jax_nadamw_full_budget.py -------------------------------------------------------------------------------- /algorithms/baselines/self_tuning/pytorch_nadamw_full_budget.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/baselines/self_tuning/pytorch_nadamw_full_budget.py -------------------------------------------------------------------------------- /algorithms/development_algorithms/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/development_algorithms/cifar/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/development_algorithms/cifar/cifar_jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/development_algorithms/cifar/cifar_jax/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/development_algorithms/cifar/cifar_jax/submission.py -------------------------------------------------------------------------------- /algorithms/development_algorithms/cifar/cifar_pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/development_algorithms/cifar/cifar_pytorch/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/development_algorithms/cifar/cifar_pytorch/submission.py -------------------------------------------------------------------------------- /algorithms/development_algorithms/cifar/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/development_algorithms/cifar/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/development_algorithms/mnist/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/development_algorithms/mnist/discrete_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/development_algorithms/mnist/discrete_space.json -------------------------------------------------------------------------------- /algorithms/development_algorithms/mnist/mnist_jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/development_algorithms/mnist/mnist_jax/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/development_algorithms/mnist/mnist_jax/submission.py -------------------------------------------------------------------------------- /algorithms/development_algorithms/mnist/mnist_pytorch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /algorithms/development_algorithms/mnist/mnist_pytorch/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/development_algorithms/mnist/mnist_pytorch/submission.py -------------------------------------------------------------------------------- /algorithms/development_algorithms/mnist/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/development_algorithms/mnist/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/submission_checker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/submission_checker.py -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/README.md -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/__init__.py -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/cosine_warmup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/cosine_warmup.py -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/criteo1tb/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/criteo1tb/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/criteo1tb_embed_init/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/criteo1tb_embed_init/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/criteo1tb_layernorm/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/criteo1tb_layernorm/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/criteo1tb_resnet/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/criteo1tb_resnet/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/data_selection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/data_selection.py -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/fastmri/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/fastmri/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/fastmri_layernorm/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/fastmri_layernorm/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/fastmri_model_size/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/fastmri_model_size/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/fastmri_tanh/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/fastmri_tanh/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/get_batch_size.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/get_batch_size.py -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/imagenet_resnet/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/imagenet_resnet/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/imagenet_resnet_gelu/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/imagenet_resnet_gelu/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/imagenet_resnet_large_bn_init/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/imagenet_resnet_large_bn_init/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/imagenet_resnet_silu/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/imagenet_resnet_silu/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/imagenet_vit/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/imagenet_vit/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/imagenet_vit_glu/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/imagenet_vit_glu/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/imagenet_vit_map/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/imagenet_vit_map/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/imagenet_vit_post_ln/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/imagenet_vit_post_ln/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/jax_adamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/jax_adamw.py -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/jax_momentum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/jax_momentum.py -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/jax_nadamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/jax_nadamw.py -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/jax_nesterov.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/jax_nesterov.py -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/jax_submission_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/jax_submission_base.py -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/librispeech_conformer/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/librispeech_conformer/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/librispeech_conformer_gelu/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/librispeech_conformer_gelu/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/librispeech_deepspeech/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/librispeech_deepspeech/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/librispeech_deepspeech_tanh/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/librispeech_deepspeech_tanh/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/ogbg/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/ogbg/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/ogbg_gelu/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/ogbg_gelu/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/ogbg_model_size/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/ogbg_model_size/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/ogbg_silu/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/ogbg_silu/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/pytorch_adamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/pytorch_adamw.py -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/pytorch_momentum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/pytorch_momentum.py -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/pytorch_nadamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/pytorch_nadamw.py -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/pytorch_nesterov.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/pytorch_nesterov.py -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/pytorch_submission_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/pytorch_submission_base.py -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/wmt/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/wmt/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/wmt_attention_temp/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/wmt_attention_temp/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/wmt_glu_tanh/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/wmt_glu_tanh/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/target_setting_algorithms/wmt_post_ln/tuning_search_space.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/target_setting_algorithms/wmt_post_ln/tuning_search_space.json -------------------------------------------------------------------------------- /algorithms/template/submission.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/algorithms/template/submission.py -------------------------------------------------------------------------------- /datasets/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/datasets/README.md -------------------------------------------------------------------------------- /datasets/dataset_setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/datasets/dataset_setup.py -------------------------------------------------------------------------------- /datasets/librispeech_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/datasets/librispeech_preprocess.py -------------------------------------------------------------------------------- /datasets/librispeech_tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/datasets/librispeech_tokenizer.py -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docker/Singularity.def: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/docker/Singularity.def -------------------------------------------------------------------------------- /docker/build_docker_images.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/docker/build_docker_images.sh -------------------------------------------------------------------------------- /docker/scripts/check_gpu.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/docker/scripts/check_gpu.py -------------------------------------------------------------------------------- /docker/scripts/cloud-init.cfg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/docker/scripts/cloud-init.cfg -------------------------------------------------------------------------------- /docker/scripts/cloud-startup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/docker/scripts/cloud-startup.sh -------------------------------------------------------------------------------- /docker/scripts/singularity_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/docker/scripts/singularity_converter.py -------------------------------------------------------------------------------- /docker/scripts/startup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/docker/scripts/startup.sh -------------------------------------------------------------------------------- /docker/scripts/test_startup.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/docker/scripts/test_startup.sh -------------------------------------------------------------------------------- /docs/CHANGELOG.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/docs/CHANGELOG.md -------------------------------------------------------------------------------- /docs/CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/docs/CONTRIBUTING.md -------------------------------------------------------------------------------- /docs/DOCUMENTATION.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/docs/DOCUMENTATION.md -------------------------------------------------------------------------------- /docs/GETTING_STARTED.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/docs/GETTING_STARTED.md -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/pyproject.toml -------------------------------------------------------------------------------- /scoring/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scoring/algoperf_v05/generate_held_out_workloads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/algoperf_v05/generate_held_out_workloads.py -------------------------------------------------------------------------------- /scoring/algoperf_v05/held_out_workloads_algoperf_v05.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/algoperf_v05/held_out_workloads_algoperf_v05.json -------------------------------------------------------------------------------- /scoring/algoperf_v05/held_out_workloads_example.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/algoperf_v05/held_out_workloads_example.json -------------------------------------------------------------------------------- /scoring/algoperf_v05/score_submissions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/algoperf_v05/score_submissions.py -------------------------------------------------------------------------------- /scoring/compute_speedups.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/compute_speedups.py -------------------------------------------------------------------------------- /scoring/performance_profile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/performance_profile.py -------------------------------------------------------------------------------- /scoring/score_submissions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/score_submissions.py -------------------------------------------------------------------------------- /scoring/scoring_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/scoring_utils.py -------------------------------------------------------------------------------- /scoring/test_data/adamw_fastmri_jax_04-18-2023-13-10-58.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/test_data/adamw_fastmri_jax_04-18-2023-13-10-58.log -------------------------------------------------------------------------------- /scoring/test_data/experiment_dir/study_0/mnist_jax/trial_0/eval_measurements.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/test_data/experiment_dir/study_0/mnist_jax/trial_0/eval_measurements.csv -------------------------------------------------------------------------------- /scoring/test_data/experiment_dir/study_0/mnist_jax/trial_1/eval_measurements.csv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/test_data/experiment_dir/study_0/mnist_jax/trial_1/eval_measurements.csv -------------------------------------------------------------------------------- /scoring/test_scoring_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/test_scoring_utils.py -------------------------------------------------------------------------------- /scoring/utils/export_runs_to_wandb.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/utils/export_runs_to_wandb.py -------------------------------------------------------------------------------- /scoring/utils/package_logs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/utils/package_logs.py -------------------------------------------------------------------------------- /scoring/utils/run_workloads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/utils/run_workloads.py -------------------------------------------------------------------------------- /scoring/utils/slurm/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/utils/slurm/README.md -------------------------------------------------------------------------------- /scoring/utils/slurm/algoperf_slurm_cluster.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/utils/slurm/algoperf_slurm_cluster.yaml -------------------------------------------------------------------------------- /scoring/utils/slurm/algoperf_slurm_pakcer_builder.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/utils/slurm/algoperf_slurm_pakcer_builder.yaml -------------------------------------------------------------------------------- /scoring/utils/slurm/config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/utils/slurm/config.json -------------------------------------------------------------------------------- /scoring/utils/slurm/make_job_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/utils/slurm/make_job_config.py -------------------------------------------------------------------------------- /scoring/utils/slurm/run_jobs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/utils/slurm/run_jobs.sh -------------------------------------------------------------------------------- /scoring/utils/workload_metadata_external_tuning.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/utils/workload_metadata_external_tuning.json -------------------------------------------------------------------------------- /scoring/utils/workload_metadata_self_tuning.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/scoring/utils/workload_metadata_self_tuning.json -------------------------------------------------------------------------------- /submission_runner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/submission_runner.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/criteo1tb/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/criteo1tb/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/criteo1tb/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/criteo1tb_embed_init/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/criteo1tb_embed_init/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/criteo1tb_embed_init/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/criteo1tb_layernorm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/criteo1tb_layernorm/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/criteo1tb_layernorm/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/criteo1tb_resnet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/criteo1tb_resnet/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/criteo1tb_resnet/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/diff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/diff.py -------------------------------------------------------------------------------- /tests/modeldiffs/fastmri/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/fastmri/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/fastmri/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/fastmri_layernorm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/fastmri_layernorm/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/fastmri_layernorm/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/fastmri_model_size/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/fastmri_model_size/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/fastmri_model_size/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/fastmri_tanh/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/fastmri_tanh/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/fastmri_tanh/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/imagenet_resnet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/imagenet_resnet/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/imagenet_resnet/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/imagenet_resnet/gelu_compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/imagenet_resnet/gelu_compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/imagenet_resnet/silu_compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/imagenet_resnet/silu_compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/imagenet_vit/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/imagenet_vit/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/imagenet_vit/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/imagenet_vit_glu/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/imagenet_vit_glu/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/imagenet_vit_glu/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/imagenet_vit_map/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/imagenet_vit_map/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/imagenet_vit_map/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/imagenet_vit_postln/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/imagenet_vit_postln/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/imagenet_vit_postln/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/librispeech_conformer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/librispeech_conformer/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/librispeech_conformer/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/librispeech_conformer_attention_temperature/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/librispeech_conformer_attention_temperature/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/librispeech_conformer_attention_temperature/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/librispeech_conformer_gelu/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/librispeech_conformer_gelu/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/librispeech_conformer_gelu/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/librispeech_conformer_layernorm/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/librispeech_conformer_layernorm/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/librispeech_conformer_layernorm/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/librispeech_deepspeech/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/librispeech_deepspeech/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/librispeech_deepspeech/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/librispeech_deepspeech_noresnet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/librispeech_deepspeech_noresnet/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/librispeech_deepspeech_noresnet/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/librispeech_deepspeech_normaug/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/librispeech_deepspeech_normaug/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/librispeech_deepspeech_normaug/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/librispeech_deepspeech_tanh/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/librispeech_deepspeech_tanh/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/librispeech_deepspeech_tanh/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/ogbg/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/ogbg/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/ogbg/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/ogbg_gelu/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/ogbg_gelu/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/ogbg_gelu/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/ogbg_model_size/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/ogbg_model_size/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/ogbg_model_size/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/ogbg_silu/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/ogbg_silu/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/ogbg_silu/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/torch2jax_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/torch2jax_utils.py -------------------------------------------------------------------------------- /tests/modeldiffs/vanilla_sgd_jax.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/vanilla_sgd_jax.py -------------------------------------------------------------------------------- /tests/modeldiffs/vanilla_sgd_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/vanilla_sgd_pytorch.py -------------------------------------------------------------------------------- /tests/modeldiffs/wmt/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/wmt/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/wmt/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/wmt_attention_temp/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/wmt_attention_temp/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/wmt_attention_temp/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/wmt_glu_tanh/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/wmt_glu_tanh/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/wmt_glu_tanh/compare.py -------------------------------------------------------------------------------- /tests/modeldiffs/wmt_post_ln/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/modeldiffs/wmt_post_ln/compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/modeldiffs/wmt_post_ln/compare.py -------------------------------------------------------------------------------- /tests/reference_algorithm_tests.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/reference_algorithm_tests.py -------------------------------------------------------------------------------- /tests/submission_runner_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/submission_runner_test.py -------------------------------------------------------------------------------- /tests/test_baselines.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/test_baselines.py -------------------------------------------------------------------------------- /tests/test_jax_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/test_jax_utils.py -------------------------------------------------------------------------------- /tests/test_num_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/test_num_params.py -------------------------------------------------------------------------------- /tests/test_param_shapes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/test_param_shapes.py -------------------------------------------------------------------------------- /tests/test_param_types.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/test_param_types.py -------------------------------------------------------------------------------- /tests/test_ssim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/test_ssim.py -------------------------------------------------------------------------------- /tests/test_traindiffs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/test_traindiffs.py -------------------------------------------------------------------------------- /tests/test_version.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/test_version.py -------------------------------------------------------------------------------- /tests/workloads/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/workloads/imagenet_resnet/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/workloads/imagenet_resnet/imagenet_jax/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/workloads/imagenet_resnet/imagenet_jax/workload_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/mlcommons/algorithmic-efficiency/HEAD/tests/workloads/imagenet_resnet/imagenet_jax/workload_test.py --------------------------------------------------------------------------------