.dockerignore
.gitignore
.pre-commit-config.yaml
CONTRIBUTING.md
LICENSE.md
README.md
RULES.md
__init__.py
setup.cfg
setup.py
submission_runner.py
.assets/mlc_logo.png
.assets/performance_profiles.png
.github/CODEOWNERS
.github/ISSUE_TEMPLATE/bug_report.md
.github/ISSUE_TEMPLATE/feature_request.md
.github/ISSUE_TEMPLATE/new_workload.md
.github/workflows/CI.yml
.github/workflows/cla.yml
.github/workflows/linting.yml
algorithmic_efficiency/__init__.py
algorithmic_efficiency/adamk.json
algorithmic_efficiency/adamqlr.json
algorithmic_efficiency/checkpoint_utils.py
algorithmic_efficiency/data_utils.py
algorithmic_efficiency/halton.py
algorithmic_efficiency/init_utils.py
algorithmic_efficiency/interop_utils.py
algorithmic_efficiency/logger_utils.py
algorithmic_efficiency/new_adamk.json
algorithmic_efficiency/old_spec.py
algorithmic_efficiency/param_utils.py
algorithmic_efficiency/profiler.py
algorithmic_efficiency/pytorch_utils.py
algorithmic_efficiency/random_utils.py
algorithmic_efficiency/spec.py
algorithmic_efficiency.egg-info/PKG-INFO
algorithmic_efficiency.egg-info/SOURCES.txt
algorithmic_efficiency.egg-info/dependency_links.txt
algorithmic_efficiency.egg-info/not-zip-safe
algorithmic_efficiency.egg-info/requires.txt
algorithmic_efficiency.egg-info/top_level.txt
algorithmic_efficiency/workloads/__init__.py
algorithmic_efficiency/workloads/cifar/__init__.py
algorithmic_efficiency/workloads/cifar/workload.py
algorithmic_efficiency/workloads/cifar/cifar_jax/__init__.py
algorithmic_efficiency/workloads/cifar/cifar_jax/input_pipeline.py
algorithmic_efficiency/workloads/cifar/cifar_jax/models.py
algorithmic_efficiency/workloads/cifar/cifar_jax/workload.py
algorithmic_efficiency/workloads/cifar/cifar_pytorch/__init__.py
algorithmic_efficiency/workloads/cifar/cifar_pytorch/models.py
algorithmic_efficiency/workloads/cifar/cifar_pytorch/workload.py
algorithmic_efficiency/workloads/criteo1tb/__init__.py
algorithmic_efficiency/workloads/criteo1tb/input_pipeline.py
algorithmic_efficiency/workloads/criteo1tb/workload.py
algorithmic_efficiency/workloads/criteo1tb/criteo1tb_jax/__init__.py
algorithmic_efficiency/workloads/criteo1tb/criteo1tb_jax/models.py
algorithmic_efficiency/workloads/criteo1tb/criteo1tb_jax/workload.py
algorithmic_efficiency/workloads/criteo1tb/criteo1tb_pytorch/__init__.py
algorithmic_efficiency/workloads/criteo1tb/criteo1tb_pytorch/models.py
algorithmic_efficiency/workloads/criteo1tb/criteo1tb_pytorch/workload.py
algorithmic_efficiency/workloads/fastmri/__init__.py
algorithmic_efficiency/workloads/fastmri/input_pipeline.py
algorithmic_efficiency/workloads/fastmri/workload.py
algorithmic_efficiency/workloads/fastmri/fastmri_jax/__init__.py
algorithmic_efficiency/workloads/fastmri/fastmri_jax/models.py
algorithmic_efficiency/workloads/fastmri/fastmri_jax/ssim.py
algorithmic_efficiency/workloads/fastmri/fastmri_jax/workload.py
algorithmic_efficiency/workloads/fastmri/fastmri_pytorch/__init__.py
algorithmic_efficiency/workloads/fastmri/fastmri_pytorch/models.py
algorithmic_efficiency/workloads/fastmri/fastmri_pytorch/ssim.py
algorithmic_efficiency/workloads/fastmri/fastmri_pytorch/workload.py
algorithmic_efficiency/workloads/imagenet_resnet/__init__.py
algorithmic_efficiency/workloads/imagenet_resnet/imagenet_v2.py
algorithmic_efficiency/workloads/imagenet_resnet/workload.py
algorithmic_efficiency/workloads/imagenet_resnet/imagenet_jax/__init__.py
algorithmic_efficiency/workloads/imagenet_resnet/imagenet_jax/input_pipeline.py
algorithmic_efficiency/workloads/imagenet_resnet/imagenet_jax/models.py
algorithmic_efficiency/workloads/imagenet_resnet/imagenet_jax/randaugment.py
algorithmic_efficiency/workloads/imagenet_resnet/imagenet_jax/workload.py
algorithmic_efficiency/workloads/imagenet_resnet/imagenet_pytorch/__init__.py
algorithmic_efficiency/workloads/imagenet_resnet/imagenet_pytorch/models.py
algorithmic_efficiency/workloads/imagenet_resnet/imagenet_pytorch/randaugment.py
algorithmic_efficiency/workloads/imagenet_resnet/imagenet_pytorch/workload.py
algorithmic_efficiency/workloads/imagenet_vit/__init__.py
algorithmic_efficiency/workloads/imagenet_vit/workload.py
algorithmic_efficiency/workloads/imagenet_vit/imagenet_jax/__init__.py
algorithmic_efficiency/workloads/imagenet_vit/imagenet_jax/models.py
algorithmic_efficiency/workloads/imagenet_vit/imagenet_jax/workload.py
algorithmic_efficiency/workloads/imagenet_vit/imagenet_pytorch/__init__.py
algorithmic_efficiency/workloads/imagenet_vit/imagenet_pytorch/models.py
algorithmic_efficiency/workloads/imagenet_vit/imagenet_pytorch/workload.py
algorithmic_efficiency/workloads/librispeech_conformer/__init__.py
algorithmic_efficiency/workloads/librispeech_conformer/input_pipeline.py
algorithmic_efficiency/workloads/librispeech_conformer/metrics.py
algorithmic_efficiency/workloads/librispeech_conformer/workload.py
algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/__init__.py
algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/librispeech_preprocessor.py
algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/models.py
algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/spectrum_augmenter.py
algorithmic_efficiency/workloads/librispeech_conformer/librispeech_jax/workload.py
algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/__init__.py
algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/models.py
algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/preprocessor.py
algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/spectrum_augmenter.py
algorithmic_efficiency/workloads/librispeech_conformer/librispeech_pytorch/workload.py
algorithmic_efficiency/workloads/librispeech_deepspeech/__init__.py
algorithmic_efficiency/workloads/librispeech_deepspeech/workload.py
algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/__init__.py
algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/models.py
algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_jax/workload.py
algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_pytorch/__init__.py
algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_pytorch/models.py
algorithmic_efficiency/workloads/librispeech_deepspeech/librispeech_pytorch/workload.py
algorithmic_efficiency/workloads/mnist/__init__.py
algorithmic_efficiency/workloads/mnist/workload.py
algorithmic_efficiency/workloads/mnist/mnist_jax/__init__.py
algorithmic_efficiency/workloads/mnist/mnist_jax/workload.py
algorithmic_efficiency/workloads/mnist/mnist_pytorch/__init__.py
algorithmic_efficiency/workloads/mnist/mnist_pytorch/workload.py
algorithmic_efficiency/workloads/ogbg/__init__.py
algorithmic_efficiency/workloads/ogbg/input_pipeline.py
algorithmic_efficiency/workloads/ogbg/metrics.py
algorithmic_efficiency/workloads/ogbg/workload.py
algorithmic_efficiency/workloads/ogbg/ogbg_jax/__init__.py
algorithmic_efficiency/workloads/ogbg/ogbg_jax/models.py
algorithmic_efficiency/workloads/ogbg/ogbg_jax/workload.py
algorithmic_efficiency/workloads/ogbg/ogbg_pytorch/__init__.py
algorithmic_efficiency/workloads/ogbg/ogbg_pytorch/models.py
algorithmic_efficiency/workloads/ogbg/ogbg_pytorch/workload.py
algorithmic_efficiency/workloads/wmt/__init__.py
algorithmic_efficiency/workloads/wmt/bleu.py
algorithmic_efficiency/workloads/wmt/input_pipeline.py
algorithmic_efficiency/workloads/wmt/tokenizer.py
algorithmic_efficiency/workloads/wmt/workload.py
algorithmic_efficiency/workloads/wmt/wmt_jax/README.md
algorithmic_efficiency/workloads/wmt/wmt_jax/__init__.py
algorithmic_efficiency/workloads/wmt/wmt_jax/decode.py
algorithmic_efficiency/workloads/wmt/wmt_jax/models.py
algorithmic_efficiency/workloads/wmt/wmt_jax/workload.py
algorithmic_efficiency/workloads/wmt/wmt_pytorch/__init__.py
algorithmic_efficiency/workloads/wmt/wmt_pytorch/decode.py
algorithmic_efficiency/workloads/wmt/wmt_pytorch/models.py
algorithmic_efficiency/workloads/wmt/wmt_pytorch/workload.py
baselines/README.md
baselines/__init__.py
baselines/adafactor/__init__.py
baselines/adafactor/tuning_search_space.json
baselines/adafactor/tuning_search_space_no_beta1.json
baselines/adafactor/jax/__init__.py
baselines/adafactor/jax/sharded_adafactor.py
baselines/adafactor/jax/submission.py
baselines/adafactor/pytorch/__init__.py
baselines/adafactor/pytorch/submission.py
baselines/adamw/__init__.py
baselines/adamw/tuning_search_space.json
baselines/adamw/tuning_search_space_no_beta1.json
baselines/adamw/jax/__init__.py
baselines/adamw/jax/submission.py
baselines/adamw/pytorch/__init__.py
baselines/adamw/pytorch/submission.py
baselines/lamb/__init__.py
baselines/lamb/tuning_search_space.json
baselines/lamb/tuning_search_space_no_beta1.json
baselines/lamb/jax/__init__.py
baselines/lamb/jax/submission.py
baselines/lamb/pytorch/__init__.py
baselines/lamb/pytorch/submission.py
baselines/momentum/__init__.py
baselines/momentum/tuning_search_space.json
baselines/momentum/tuning_search_space_no_beta1.json
baselines/momentum/jax/__init__.py
baselines/momentum/jax/submission.py
baselines/momentum/pytorch/__init__.py
baselines/momentum/pytorch/submission.py
baselines/nadamw/__init__.py
baselines/nadamw/tuning_search_space.json
baselines/nadamw/tuning_search_space_no_beta1.json
baselines/nadamw/jax/__init__.py
baselines/nadamw/jax/submission.py
baselines/nadamw/pytorch/__init__.py
baselines/nadamw/pytorch/submission.py
baselines/nesterov/__init__.py
baselines/nesterov/tuning_search_space.json
baselines/nesterov/tuning_search_space_no_beta1.json
baselines/nesterov/jax/__init__.py
baselines/nesterov/jax/submission.py
baselines/nesterov/pytorch/__init__.py
baselines/nesterov/pytorch/submission.py
baselines/sam/__init__.py
baselines/sam/tuning_search_space.json
baselines/sam/tuning_search_space_no_beta1.json
baselines/sam/jax/__init__.py
baselines/sam/jax/submission.py
baselines/sam/pytorch/__init__.py
baselines/sam/pytorch/submission.py
baselines/shampoo/__init__.py
baselines/shampoo/tuning_search_space.json
baselines/shampoo/tuning_search_space_no_beta1.json
baselines/shampoo/jax/__init__.py
baselines/shampoo/jax/distributed_shampoo.py
baselines/shampoo/jax/submission.py
baselines/shampoo/pytorch/__init__.py
datasets/README.md
datasets/dataset_setup.py
datasets/librispeech_preprocess.py
datasets/librispeech_tokenizer.py
docker/Dockerfile
docker/README.md
docker/scripts/check_gpu.py
docker/scripts/cloud-init.cfg
reference_algorithms/__init__.py
reference_algorithms/development_algorithms/README.md
reference_algorithms/development_algorithms/__init__.py
reference_algorithms/development_algorithms/cifar/__init__.py
reference_algorithms/development_algorithms/cifar/tuning_search_space.json
reference_algorithms/development_algorithms/cifar/cifar_jax/__init__.py
reference_algorithms/development_algorithms/cifar/cifar_jax/submission.py
reference_algorithms/development_algorithms/cifar/cifar_pytorch/__init__.py
reference_algorithms/development_algorithms/cifar/cifar_pytorch/submission.py
reference_algorithms/development_algorithms/criteo1tb/tuning_search_space.json
reference_algorithms/development_algorithms/criteo1tb/criteo1tb_jax/__init__.py
reference_algorithms/development_algorithms/criteo1tb/criteo1tb_jax/submission.py
reference_algorithms/development_algorithms/criteo1tb/criteo1tb_pytorch/__init__.py
reference_algorithms/development_algorithms/criteo1tb/criteo1tb_pytorch/submission.py
reference_algorithms/development_algorithms/fastmri/__init__.py
reference_algorithms/development_algorithms/fastmri/tuning_search_space.json
reference_algorithms/development_algorithms/fastmri/fastmri_jax/__init__.py
reference_algorithms/development_algorithms/fastmri/fastmri_jax/submission.py
reference_algorithms/development_algorithms/fastmri/fastmri_pytorch/__init__.py
reference_algorithms/development_algorithms/fastmri/fastmri_pytorch/submission.py
reference_algorithms/development_algorithms/imagenet_resnet/__init__.py
reference_algorithms/development_algorithms/imagenet_resnet/tuning_search_space.json
reference_algorithms/development_algorithms/imagenet_resnet/imagenet_jax/__init__.py
reference_algorithms/development_algorithms/imagenet_resnet/imagenet_jax/submission.py
reference_algorithms/development_algorithms/imagenet_resnet/imagenet_pytorch/__init__.py
reference_algorithms/development_algorithms/imagenet_resnet/imagenet_pytorch/submission.py
reference_algorithms/development_algorithms/imagenet_vit/__init__.py
reference_algorithms/development_algorithms/imagenet_vit/tuning_search_space.json
reference_algorithms/development_algorithms/imagenet_vit/imagenet_jax/__init__.py
reference_algorithms/development_algorithms/imagenet_vit/imagenet_jax/submission.py
reference_algorithms/development_algorithms/imagenet_vit/imagenet_pytorch/__init__.py
reference_algorithms/development_algorithms/imagenet_vit/imagenet_pytorch/submission.py
reference_algorithms/development_algorithms/librispeech_conformer/__init__.py
reference_algorithms/development_algorithms/librispeech_conformer/tuning_search_space.json
reference_algorithms/development_algorithms/librispeech_conformer/librispeech_jax/__init__.py
reference_algorithms/development_algorithms/librispeech_conformer/librispeech_jax/submission.py
reference_algorithms/development_algorithms/librispeech_conformer/librispeech_pytorch/__init__.py
reference_algorithms/development_algorithms/librispeech_conformer/librispeech_pytorch/submission.py
reference_algorithms/development_algorithms/librispeech_deepspeech/__init__.py
reference_algorithms/development_algorithms/librispeech_deepspeech/tuning_search_space.json
reference_algorithms/development_algorithms/librispeech_deepspeech/librispeech_jax/__init__.py
reference_algorithms/development_algorithms/librispeech_deepspeech/librispeech_jax/submission.py
reference_algorithms/development_algorithms/librispeech_deepspeech/librispeech_pytorch/__init__.py
reference_algorithms/development_algorithms/librispeech_deepspeech/librispeech_pytorch/submission.py
reference_algorithms/development_algorithms/mnist/__init__.py
reference_algorithms/development_algorithms/mnist/discrete_space.json
reference_algorithms/development_algorithms/mnist/tuning_search_space.json
reference_algorithms/development_algorithms/mnist/mnist_jax/__init__.py
reference_algorithms/development_algorithms/mnist/mnist_jax/submission.py
reference_algorithms/development_algorithms/mnist/mnist_pytorch/__init__.py
reference_algorithms/development_algorithms/mnist/mnist_pytorch/submission.py
reference_algorithms/development_algorithms/ogbg/__init__.py
reference_algorithms/development_algorithms/ogbg/tuning_search_space.json
reference_algorithms/development_algorithms/ogbg/ogbg_jax/__init__.py
reference_algorithms/development_algorithms/ogbg/ogbg_jax/submission.py
reference_algorithms/development_algorithms/ogbg/ogbg_pytorch/__init__.py
reference_algorithms/development_algorithms/ogbg/ogbg_pytorch/submission.py
reference_algorithms/development_algorithms/wmt/__init__.py
reference_algorithms/development_algorithms/wmt/tuning_search_space.json
reference_algorithms/development_algorithms/wmt/wmt_jax/__init__.py
reference_algorithms/development_algorithms/wmt/wmt_jax/submission.py
reference_algorithms/development_algorithms/wmt/wmt_pytorch/__init__.py
reference_algorithms/development_algorithms/wmt/wmt_pytorch/submission.py
reference_algorithms/target_setting_algorithms/README.md
reference_algorithms/target_setting_algorithms/__init__.py
reference_algorithms/target_setting_algorithms/cosine_warmup.py
reference_algorithms/target_setting_algorithms/data_selection.py
reference_algorithms/target_setting_algorithms/get_batch_size.py
reference_algorithms/target_setting_algorithms/jax_adamw.py
reference_algorithms/target_setting_algorithms/jax_momentum.py
reference_algorithms/target_setting_algorithms/jax_nadamw.py
reference_algorithms/target_setting_algorithms/jax_nesterov.py
reference_algorithms/target_setting_algorithms/jax_submission_base.py
reference_algorithms/target_setting_algorithms/pytorch_adamw.py
reference_algorithms/target_setting_algorithms/pytorch_momentum.py
reference_algorithms/target_setting_algorithms/pytorch_nadamw.py
reference_algorithms/target_setting_algorithms/pytorch_nesterov.py
reference_algorithms/target_setting_algorithms/pytorch_submission_base.py
reference_algorithms/target_setting_algorithms/criteo1tb/tuning_search_space.json
reference_algorithms/target_setting_algorithms/fastmri/tuning_search_space.json
reference_algorithms/target_setting_algorithms/imagenet_resnet/tuning_search_space.json
reference_algorithms/target_setting_algorithms/imagenet_vit/tuning_search_space.json
reference_algorithms/target_setting_algorithms/librispeech_conformer/tuning_search_space.json
reference_algorithms/target_setting_algorithms/librispeech_deepspeech/tuning_search_space.json
reference_algorithms/target_setting_algorithms/ogbg/tuning_search_space.json
reference_algorithms/target_setting_algorithms/wmt/tuning_search_space.json
scoring/__init__.py
scoring/scoring.py
scoring/scoring_utils.py
scoring/test_scoring_utils.py
scoring/test_data/adamw_fastmri_jax_04-18-2023-13-10-58.log
tests/__init__.py
tests/reference_algorithm_tests.py
tests/submission_runner_test.py
tests/test_baselines.py
tests/test_num_params.py
tests/test_param_shapes.py
tests/test_param_types.py
tests/test_ssim.py
tests/test_traindiffs.py
tests/version_test.py
tests/modeldiffs/__init__.py
tests/modeldiffs/diff.py
tests/modeldiffs/torch2jax_utils.py
tests/modeldiffs/vanilla_sgd_jax.py
tests/modeldiffs/vanilla_sgd_pytorch.py
tests/modeldiffs/fastmri/__init__.py
tests/modeldiffs/fastmri/compare.py
tests/modeldiffs/imagenet_resnet/__init__.py
tests/modeldiffs/imagenet_resnet/compare.py
tests/modeldiffs/imagenet_vit/__init__.py
tests/modeldiffs/imagenet_vit/compare.py
tests/modeldiffs/librispeech_conformer/__init__.py
tests/modeldiffs/librispeech_conformer/compare.py
tests/modeldiffs/librispeech_deepspeech/__init__.py
tests/modeldiffs/librispeech_deepspeech/compare.py
tests/modeldiffs/ogbg/__init__.py
tests/modeldiffs/ogbg/compare.py
tests/modeldiffs/wmt/__init__.py
tests/modeldiffs/wmt/compare.py
tests/workloads/__init__.py
tests/workloads/imagenet_resnet/__init__.py
tests/workloads/imagenet_resnet/imagenet_jax/__init__.py
tests/workloads/imagenet_resnet/imagenet_jax/workload_test.py