LICENSE
README.md
setup.py
learned_optimization/__init__.py
learned_optimization/checkpoints.py
learned_optimization/distributed.py
learned_optimization/distributed_test.py
learned_optimization/eval_training.py
learned_optimization/eval_training_test.py
learned_optimization/filesystem.py
learned_optimization/outer_train.py
learned_optimization/outer_train_test.py
learned_optimization/profile.py
learned_optimization/setup_experiment.py
learned_optimization/summary.py
learned_optimization/summary_test.py
learned_optimization/training.py
learned_optimization/tree_utils.py
learned_optimization.egg-info/PKG-INFO
learned_optimization.egg-info/SOURCES.txt
learned_optimization.egg-info/dependency_links.txt
learned_optimization.egg-info/not-zip-safe
learned_optimization.egg-info/requires.txt
learned_optimization.egg-info/top_level.txt
learned_optimization/baselines/__init__.py
learned_optimization/baselines/hparam_sets.py
learned_optimization/baselines/run_archive.py
learned_optimization/baselines/run_trainer.py
learned_optimization/baselines/run_trainer_test.py
learned_optimization/baselines/utils.py
learned_optimization/continuous_eval/__init__.py
learned_optimization/continuous_eval/run_eval_chief.py
learned_optimization/continuous_eval/run_eval_chief_test.py
learned_optimization/continuous_eval/run_eval_worker.py
learned_optimization/continuous_eval/run_eval_worker_test.py
learned_optimization/continuous_eval/task_group_server.py
learned_optimization/continuous_eval/task_group_server_test.py
learned_optimization/learned_optimizers/__init__.py
learned_optimization/learned_optimizers/adafac_mlp_lopt.py
learned_optimization/learned_optimizers/adafac_mlp_lopt_test.py
learned_optimization/learned_optimizers/base.py
learned_optimization/learned_optimizers/base_test.py
learned_optimization/learned_optimizers/common.py
learned_optimization/learned_optimizers/mlp_lopt.py
learned_optimization/learned_optimizers/mlp_lopt_test.py
learned_optimization/learned_optimizers/nn_adam.py
learned_optimization/learned_optimizers/nn_adam_test.py
learned_optimization/learned_optimizers/rnn_mlp_lopt.py
learned_optimization/learned_optimizers/rnn_mlp_lopt_test.py
learned_optimization/learned_optimizers/test_utils.py
learned_optimization/optimizers/__init__.py
learned_optimization/optimizers/base.py
learned_optimization/optimizers/base_test.py
learned_optimization/optimizers/test_utils.py
learned_optimization/outer_trainers/__init__.py
learned_optimization/outer_trainers/common.py
learned_optimization/outer_trainers/full_es.py
learned_optimization/outer_trainers/full_es_test.py
learned_optimization/outer_trainers/gradient_learner.py
learned_optimization/outer_trainers/gradient_learner_test.py
learned_optimization/outer_trainers/test_utils.py
learned_optimization/outer_trainers/truncated_es.py
learned_optimization/outer_trainers/truncated_es_test.py
learned_optimization/outer_trainers/truncated_grad.py
learned_optimization/outer_trainers/truncated_grad_test.py
learned_optimization/outer_trainers/truncated_pes.py
learned_optimization/outer_trainers/truncated_pes_pmap_test.py
learned_optimization/outer_trainers/truncated_pes_test.py
learned_optimization/outer_trainers/truncation_schedule.py
learned_optimization/outer_trainers/truncation_schedule_test.py
learned_optimization/population/__init__.py
learned_optimization/population/population.py
learned_optimization/population/population_test.py
learned_optimization/population/examples/__init__.py
learned_optimization/population/examples/synthetic/__init__.py
learned_optimization/population/examples/synthetic/train.py
learned_optimization/population/examples/synthetic/train_test.py
learned_optimization/population/mutators/__init__.py
learned_optimization/population/mutators/fixed_schedule.py
learned_optimization/population/mutators/fixed_schedule_test.py
learned_optimization/population/mutators/single_worker_explore.py
learned_optimization/population/mutators/single_worker_explore_test.py
learned_optimization/population/mutators/winner_take_all_genetic.py
learned_optimization/population/mutators/winner_take_all_genetic_test.py
learned_optimization/tasks/__init__.py
learned_optimization/tasks/base.py
learned_optimization/tasks/es_wrapper.py
learned_optimization/tasks/es_wrapper_test.py
learned_optimization/tasks/generative_model_utils.py
learned_optimization/tasks/quadratics.py
learned_optimization/tasks/quadratics_test.py
learned_optimization/tasks/resnet.py
learned_optimization/tasks/rnn.py
learned_optimization/tasks/task_augmentation.py
learned_optimization/tasks/task_augmentation_test.py
learned_optimization/tasks/test_utils.py
learned_optimization/tasks/transformer.py
learned_optimization/tasks/datasets/__init__.py
learned_optimization/tasks/datasets/base.py
learned_optimization/tasks/datasets/image.py
learned_optimization/tasks/datasets/image_test.py
learned_optimization/tasks/datasets/language.py
learned_optimization/tasks/datasets/language_test.py
learned_optimization/tasks/fixed/__init__.py
learned_optimization/tasks/fixed/conv.py
learned_optimization/tasks/fixed/conv_test.py
learned_optimization/tasks/fixed/es_wrapped.py
learned_optimization/tasks/fixed/es_wrapped_test.py
learned_optimization/tasks/fixed/image_mlp.py
learned_optimization/tasks/fixed/image_mlp_ae.py
learned_optimization/tasks/fixed/image_mlp_ae_test.py
learned_optimization/tasks/fixed/image_mlp_test.py
learned_optimization/tasks/fixed/resnet.py
learned_optimization/tasks/fixed/resnet_test.py
learned_optimization/tasks/fixed/rnn_lm.py
learned_optimization/tasks/fixed/rnn_lm_test.py
learned_optimization/tasks/fixed/transformer_lm.py
learned_optimization/tasks/fixed/transformer_lm_test.py
learned_optimization/tasks/parametric/__init__.py
learned_optimization/tasks/parametric/cfgobject.py
learned_optimization/tasks/parametric/cfgobject_test.py
learned_optimization/tasks/parametric/image_conv.py
learned_optimization/tasks/parametric/image_conv_test.py
learned_optimization/tasks/parametric/image_mlp.py
learned_optimization/tasks/parametric/image_mlp_ae.py
learned_optimization/tasks/parametric/image_mlp_ae_test.py
learned_optimization/tasks/parametric/image_mlp_test.py
learned_optimization/tasks/parametric/image_mlp_vae.py
learned_optimization/tasks/parametric/image_mlp_vae_test.py
learned_optimization/tasks/parametric/image_resnet.py
learned_optimization/tasks/parametric/image_resnet_test.py
learned_optimization/tasks/parametric/lm_rnn.py
learned_optimization/tasks/parametric/lm_rnn_test.py
learned_optimization/tasks/parametric/lm_transformer.py
learned_optimization/tasks/parametric/lm_transformer_test.py
learned_optimization/tasks/parametric/parametric_utils.py
learned_optimization/tasks/parametric/parametric_utils_test.py