CHANGELOG.md
LICENSE
MANIFEST.in
README.md
setup.cfg
setup.py
docs/Makefile
docs/aggregation.rst
docs/api.rst
docs/clientapi.rst
docs/conf.py
docs/controlflow.rst
docs/dataset.rst
docs/evaluation.rst
docs/examples.rst
docs/index.rst
docs/lr_scheduler.rst
docs/models.rst
docs/optim.rst
docs/utils.rst
mlbench_core/__init__.py
mlbench_core/api.py
mlbench_core/install_cuda_extensions.py
mlbench_core.egg-info/PKG-INFO
mlbench_core.egg-info/SOURCES.txt
mlbench_core.egg-info/dependency_links.txt
mlbench_core.egg-info/entry_points.txt
mlbench_core.egg-info/not-zip-safe
mlbench_core.egg-info/requires.txt
mlbench_core.egg-info/top_level.txt
mlbench_core/aggregation/__init__.py
mlbench_core/aggregation/pytorch/__init__.py
mlbench_core/aggregation/pytorch/aggregation.py
mlbench_core/aggregation/pytorch/centralized.py
mlbench_core/aggregation/pytorch/decentralized.py
mlbench_core/cli/__init__.py
mlbench_core/cli/aws_utils.py
mlbench_core/cli/chartbuilder.py
mlbench_core/cli/cli.py
mlbench_core/cli/gcloud_utils.py
mlbench_core/cli/kind_utils.py
mlbench_core/cli/utils.py
mlbench_core/controlflow/__init__.py
mlbench_core/controlflow/pytorch/__init__.py
mlbench_core/controlflow/pytorch/checkpoints_evaluation.py
mlbench_core/controlflow/pytorch/controlflow.py
mlbench_core/controlflow/pytorch/helpers.py
mlbench_core/controlflow/pytorch/train_validation.py
mlbench_core/controlflow/tensorflow/__init__.py
mlbench_core/controlflow/tensorflow/train_validation.py
mlbench_core/dataset/__init__.py
mlbench_core/dataset/imagerecognition/__init__.py
mlbench_core/dataset/imagerecognition/pytorch/__init__.py
mlbench_core/dataset/imagerecognition/pytorch/dataloader.py
mlbench_core/dataset/imagerecognition/tensorflow/__init__.py
mlbench_core/dataset/imagerecognition/tensorflow/cifar10.py
mlbench_core/dataset/linearmodels/__init__.py
mlbench_core/dataset/linearmodels/pytorch/__init__.py
mlbench_core/dataset/linearmodels/pytorch/dataloader.py
mlbench_core/dataset/nlp/__init__.py
mlbench_core/dataset/nlp/pytorch/__init__.py
mlbench_core/dataset/nlp/pytorch/wikitext2_dataset.py
mlbench_core/dataset/nlp/pytorch/wmt16_dataset.py
mlbench_core/dataset/nlp/pytorch/wmt17_dataset.py
mlbench_core/dataset/nlp/pytorch/wmt16/__init__.py
mlbench_core/dataset/nlp/pytorch/wmt16/utils.py
mlbench_core/dataset/nlp/pytorch/wmt16/wmt16_config.py
mlbench_core/dataset/nlp/pytorch/wmt16/wmt16_tokenizer.py
mlbench_core/dataset/nlp/pytorch/wmt17/__init__.py
mlbench_core/dataset/nlp/pytorch/wmt17/batching.py
mlbench_core/dataset/nlp/pytorch/wmt17/collate.py
mlbench_core/dataset/nlp/pytorch/wmt17/wmt17_dictionary.py
mlbench_core/dataset/nlp/pytorch/wmt17/preprocess/__init__.py
mlbench_core/dataset/nlp/pytorch/wmt17/preprocess/indexed_dataset.py
mlbench_core/dataset/nlp/pytorch/wmt17/preprocess/preprocess.py
mlbench_core/dataset/nlp/pytorch/wmt17/preprocess/sub_tokenizer.py
mlbench_core/dataset/util/__init__.py
mlbench_core/dataset/util/tools.py
mlbench_core/dataset/util/pytorch/__init__.py
mlbench_core/dataset/util/pytorch/libsvm.py
mlbench_core/dataset/util/pytorch/partition.py
mlbench_core/evaluation/__init__.py
mlbench_core/evaluation/goals.py
mlbench_core/evaluation/pytorch/__init__.py
mlbench_core/evaluation/pytorch/criterion.py
mlbench_core/evaluation/pytorch/metrics.py
mlbench_core/evaluation/tensorflow/__init__.py
mlbench_core/evaluation/tensorflow/criterion.py
mlbench_core/evaluation/tensorflow/metrics.py
mlbench_core/lr_scheduler/__init__.py
mlbench_core/lr_scheduler/pytorch/__init__.py
mlbench_core/lr_scheduler/pytorch/lr.py
mlbench_core/lr_scheduler/tensorflow/__init__.py
mlbench_core/lr_scheduler/tensorflow/lr.py
mlbench_core/models/__init__.py
mlbench_core/models/pytorch/__init__.py
mlbench_core/models/pytorch/linear_models.py
mlbench_core/models/pytorch/resnet.py
mlbench_core/models/pytorch/vgg.py
mlbench_core/models/pytorch/gnmt/__init__.py
mlbench_core/models/pytorch/gnmt/attention.py
mlbench_core/models/pytorch/gnmt/decoder.py
mlbench_core/models/pytorch/gnmt/encoder.py
mlbench_core/models/pytorch/gnmt/models.py
mlbench_core/models/pytorch/gnmt/translator.py
mlbench_core/models/pytorch/gnmt/utils.py
mlbench_core/models/pytorch/gnmt/attn_score/attn_score_cuda.cpp
mlbench_core/models/pytorch/gnmt/attn_score/attn_score_cuda_kernel.cu
mlbench_core/models/pytorch/language_models/__init__.py
mlbench_core/models/pytorch/language_models/lstm.py
mlbench_core/models/pytorch/layers/__init__.py
mlbench_core/models/pytorch/layers/dropout_layers.py
mlbench_core/models/pytorch/transformer/__init__.py
mlbench_core/models/pytorch/transformer/decoder.py
mlbench_core/models/pytorch/transformer/encoder.py
mlbench_core/models/pytorch/transformer/sequence_generator.py
mlbench_core/models/pytorch/transformer/transformer.py
mlbench_core/models/pytorch/transformer/modules/__init__.py
mlbench_core/models/pytorch/transformer/modules/embeddings.py
mlbench_core/models/pytorch/transformer/modules/layers.py
mlbench_core/models/pytorch/transformer/modules/multihead_attention.py
mlbench_core/models/pytorch/transformer/modules/strided_batched_gemm/strided_batched_gemm.cpp
mlbench_core/models/pytorch/transformer/modules/strided_batched_gemm/strided_batched_gemm_cuda.cu
mlbench_core/models/tensorflow/__init__.py
mlbench_core/models/tensorflow/resnet_model.py
mlbench_core/optim/__init__.py
mlbench_core/optim/pytorch/__init__.py
mlbench_core/optim/pytorch/centralized.py
mlbench_core/optim/pytorch/decentralized.py
mlbench_core/optim/pytorch/fp_optimizers.py
mlbench_core/optim/pytorch/optim.py
mlbench_core/utils/__init__.py
mlbench_core/utils/log_metrics.py
mlbench_core/utils/task_args.py
mlbench_core/utils/tracker.py
mlbench_core/utils/pytorch/__init__.py
mlbench_core/utils/pytorch/checkpoint.py
mlbench_core/utils/pytorch/distributed.py
mlbench_core/utils/pytorch/helpers.py
mlbench_core/utils/pytorch/topology.py
mlbench_core/utils/pytorch/utils.py
mlbench_core/utils/pytorch/inference/__init__.py
mlbench_core/utils/pytorch/inference/beam_search.py
mlbench_core/utils/tensorflow/__init__.py
tests/__init__.py
tests/test_api.py
tests/test_cli.py
tests/test_gcloud_cli.py
tests/test_python_optim.py
tests/test_pytorch_controlflow.py
tests/test_pytorch_helpers.py
tests/test_pytorch_metrics.py
tests/test_pytorch_models.py
tests/test_pytorch_schedulers.py
tests/test_pytorch_utils.py
tests/test_utils.py