Path	Lines of Code
ax/__init__.py	64
ax/benchmark/__init__.py	1
ax/benchmark/benchmark.py	450
ax/benchmark/benchmark_problem.py	126
ax/benchmark/benchmark_result.py	274
ax/benchmark/botorch_methods.py	122
ax/benchmark/botorch_modular/__init__.py	1
ax/benchmark/botorch_modular/standard_methods.py	64
ax/benchmark/botorch_modular/standard_problems.py	25
ax/benchmark/utils.py	26
ax/benchmark2/__init__.py	23
ax/benchmark2/benchmark.py	45
ax/benchmark2/benchmark_method.py	14
ax/benchmark2/benchmark_problem.py	114
ax/benchmark2/benchmark_result.py	109
ax/core/__init__.py	62
ax/core/arm.py	71
ax/core/base_trial.py	378
ax/core/batch_trial.py	325
ax/core/data.py	214
ax/core/experiment.py	678
ax/core/generator_run.py	235
ax/core/map_data.py	185
ax/core/map_metric.py	6
ax/core/metric.py	139
ax/core/multi_type_experiment.py	145
ax/core/objective.py	139
ax/core/observation.py	326
ax/core/optimization_config.py	313
ax/core/outcome_constraint.py	161
ax/core/parameter.py	419
ax/core/parameter_constraint.py	145
ax/core/parameter_distribution.py	34
ax/core/runner.py	44
ax/core/search_space.py	422
ax/core/trial.py	131
ax/core/types.py	59
ax/core/utils.py	144
ax/early_stopping/__init__.py	2
ax/early_stopping/strategies/__init__.py	14
ax/early_stopping/strategies/base.py	182
ax/early_stopping/strategies/percentile.py	134
ax/early_stopping/strategies/threshold.py	99
ax/early_stopping/utils.py	58
ax/exceptions/__init__.py	1
ax/exceptions/constants.py	9
ax/exceptions/core.py	56
ax/exceptions/data_provider.py	17
ax/exceptions/generation_strategy.py	15
ax/exceptions/model.py	5
ax/exceptions/storage.py	13
ax/global_stopping/__init__.py	2
ax/global_stopping/strategies/__init__.py	6
ax/global_stopping/strategies/base.py	14
ax/global_stopping/strategies/improvement.py	151
ax/metrics/__init__.py	10
ax/metrics/botorch_test_problem.py	32
ax/metrics/branin.py	15
ax/metrics/branin_map.py	114
ax/metrics/chemistry.py	71
ax/metrics/curve.py	223
ax/metrics/factorial.py	85
ax/metrics/hartmann6.py	10
ax/metrics/l2norm.py	5
ax/metrics/noisy_function.py	85
ax/metrics/noisy_function_map.py	63
ax/metrics/sklearn.py	124
ax/metrics/tensorboard.py	76
ax/metrics/torchx.py	43
ax/modelbridge/__init__.py	25
ax/modelbridge/array.py	527
ax/modelbridge/base.py	729
ax/modelbridge/cross_validation.py	244
ax/modelbridge/discrete.py	172
ax/modelbridge/dispatch_utils.py	314
ax/modelbridge/factory.py	519
ax/modelbridge/generation_node.py	145
ax/modelbridge/generation_strategy.py	474
ax/modelbridge/model_spec.py	192
ax/modelbridge/modelbridge_utils.py	675
ax/modelbridge/numpy.py	7
ax/modelbridge/random.py	75
ax/modelbridge/registry.py	413
ax/modelbridge/strategies/__init__.py	1
ax/modelbridge/strategies/alebo.py	98
ax/modelbridge/strategies/rembo.py	204
ax/modelbridge/torch.py	430
ax/modelbridge/transforms/__init__.py	1
ax/modelbridge/transforms/base.py	83
ax/modelbridge/transforms/cap_parameter.py	33
ax/modelbridge/transforms/cast.py	51
ax/modelbridge/transforms/centered_unit_x.py	59
ax/modelbridge/transforms/choice_encode.py	134
ax/modelbridge/transforms/convert_metric_names.py	105
ax/modelbridge/transforms/derelativize.py	57
ax/modelbridge/transforms/int_range_to_choice.py	51
ax/modelbridge/transforms/int_to_float.py	103
ax/modelbridge/transforms/inverse_gaussian_cdf_y.py	43
ax/modelbridge/transforms/ivw.py	70
ax/modelbridge/transforms/log.py	52
ax/modelbridge/transforms/log_y.py	133
ax/modelbridge/transforms/logit.py	52
ax/modelbridge/transforms/one_hot.py	116
ax/modelbridge/transforms/percentile_y.py	70
ax/modelbridge/transforms/power_transform_y.py	124
ax/modelbridge/transforms/relativize.py	161
ax/modelbridge/transforms/remove_fixed.py	54
ax/modelbridge/transforms/rounding.py	46
ax/modelbridge/transforms/search_space_to_choice.py	67
ax/modelbridge/transforms/standardize_y.py	92
ax/modelbridge/transforms/stratified_standardize_y.py	99
ax/modelbridge/transforms/task_encode.py	64
ax/modelbridge/transforms/trial_as_task.py	81
ax/modelbridge/transforms/unit_x.py	73
ax/modelbridge/transforms/utils.py	63
ax/modelbridge/transforms/winsorize.py	318
ax/models/__init__.py	3
ax/models/base.py	14
ax/models/discrete/__init__.py	1
ax/models/discrete/eb_thompson.py	34
ax/models/discrete/full_factorial.py	48
ax/models/discrete/thompson.py	194
ax/models/discrete_base.py	47
ax/models/model_utils.py	289
ax/models/numpy/__init__.py	1
ax/models/numpy/randomforest.py	83
ax/models/numpy_base.py	61
ax/models/random/__init__.py	1
ax/models/random/alebo_initializer.py	50
ax/models/random/base.py	151
ax/models/random/rembo_initializer.py	40
ax/models/random/sobol.py	93
ax/models/random/uniform.py	10
ax/models/torch/__init__.py	1
ax/models/torch/alebo.py	518
ax/models/torch/botorch.py	356
ax/models/torch/botorch_defaults.py	434
ax/models/torch/botorch_kg.py	320
ax/models/torch/botorch_mes.py	214
ax/models/torch/botorch_modular/__init__.py	1
ax/models/torch/botorch_modular/acquisition.py	282
ax/models/torch/botorch_modular/list_surrogate.py	106
ax/models/torch/botorch_modular/model.py	324
ax/models/torch/botorch_modular/multi_fidelity.py	70
ax/models/torch/botorch_modular/optimizer_argparse.py	94
ax/models/torch/botorch_modular/surrogate.py	256
ax/models/torch/botorch_modular/utils.py	120
ax/models/torch/botorch_moo.py	236
ax/models/torch/botorch_moo_defaults.py	300
ax/models/torch/cbo_lcea.py	139
ax/models/torch/cbo_lcem.py	71
ax/models/torch/cbo_sac.py	84
ax/models/torch/frontier_utils.py	10
ax/models/torch/fully_bayesian.py	469
ax/models/torch/fully_bayesian_model_utils.py	173
ax/models/torch/posterior_mean.py	42
ax/models/torch/rembo.py	188
ax/models/torch/utils.py	363
ax/models/torch_base.py	80
ax/models/types.py	9
ax/plot/__init__.py	1
ax/plot/bandit_rollout.py	56
ax/plot/base.py	50
ax/plot/benchmark.py	114
ax/plot/color.py	92
ax/plot/contour.py	688
ax/plot/css/__init__.py	1
ax/plot/css/base.css	41
ax/plot/diagnostic.py	398
ax/plot/feature_importances.py	175
ax/plot/helper.py	616
ax/plot/js/__init__.py	1
ax/plot/js/common/__init__.py	1
ax/plot/js/common/css.js	4
ax/plot/js/common/helpers.js	189
ax/plot/js/common/plotly_offline.js	8
ax/plot/js/common/plotly_online.js	10
ax/plot/js/common/plotly_requires.js	5
ax/plot/js/generic_plotly.js	6
ax/plot/marginal_effects.py	45
ax/plot/parallel_coordinates.py	38
ax/plot/pareto_frontier.py	795
ax/plot/pareto_utils.py	320
ax/plot/render.py	78
ax/plot/scatter.py	1246
ax/plot/slice.py	410
ax/plot/table_view.py	103
ax/plot/trace.py	343
ax/runners/__init__.py	3
ax/runners/botorch_test_problem.py	21
ax/runners/simulated_backend.py	24
ax/runners/synthetic.py	16
ax/runners/torchx.py	82
ax/service/__init__.py	2
ax/service/ax_client.py	1024
ax/service/managed_loop.py	250
ax/service/scheduler.py	872
ax/service/utils/__init__.py	1
ax/service/utils/best_point.py	454
ax/service/utils/best_point_mixin.py	153
ax/service/utils/early_stopping.py	15
ax/service/utils/instantiation.py	743
ax/service/utils/report_utils.py	420
ax/service/utils/with_db_settings_base.py	386
ax/storage/__init__.py	2
ax/storage/botorch_modular_registry.py	118
ax/storage/json_store/__init__.py	3
ax/storage/json_store/decoder.py	591
ax/storage/json_store/decoders.py	151
ax/storage/json_store/encoder.py	155
ax/storage/json_store/encoders.py	431
ax/storage/json_store/load.py	20
ax/storage/json_store/registry.py	242
ax/storage/json_store/save.py	29
ax/storage/metric_registry.py	80
ax/storage/registry_bundle.py	111
ax/storage/runner_registry.py	61
ax/storage/sqa_store/__init__.py	5
ax/storage/sqa_store/db.py	112
ax/storage/sqa_store/decoder.py	866
ax/storage/sqa_store/delete.py	7
ax/storage/sqa_store/encoder.py	719
ax/storage/sqa_store/json.py	33
ax/storage/sqa_store/load.py	332
ax/storage/sqa_store/reduced_state.py	22
ax/storage/sqa_store/save.py	356
ax/storage/sqa_store/sqa_classes.py	257
ax/storage/sqa_store/sqa_config.py	82
ax/storage/sqa_store/sqa_enum.py	38
ax/storage/sqa_store/structs.py	9
ax/storage/sqa_store/timestamp.py	17
ax/storage/sqa_store/utils.py	91
ax/storage/sqa_store/validation.py	75
ax/storage/transform_registry.py	47
ax/storage/utils.py	19
ax/utils/__init__.py	1
ax/utils/common/__init__.py	1
ax/utils/common/base.py	24
ax/utils/common/constants.py	51
ax/utils/common/docutils.py	11
ax/utils/common/equality.py	108
ax/utils/common/executils.py	132
ax/utils/common/kwargs.py	62
ax/utils/common/logger.py	85
ax/utils/common/serialization.py	76
ax/utils/common/testutils.py	197
ax/utils/common/timeutils.py	22
ax/utils/common/typeutils.py	66
ax/utils/flake8_plugins/__init__.py	1
ax/utils/flake8_plugins/docstring_checker.py	80
ax/utils/measurement/__init__.py	1
ax/utils/measurement/synthetic_functions.py	216
ax/utils/notebook/__init__.py	1
ax/utils/notebook/plotting.py	22
ax/utils/report/__init__.py	1
ax/utils/report/render.py	62
ax/utils/report/resources/__init__.py	1
ax/utils/report/resources/base_template.html	34
ax/utils/report/resources/report.css	174
ax/utils/report/resources/simple_template.html	6
ax/utils/report/resources/sufficient_statistic.html	22
ax/utils/stats/__init__.py	1
ax/utils/stats/statstools.py	180
ax/utils/testing/__init__.py	1
ax/utils/testing/backend_scheduler.py	75
ax/utils/testing/backend_simulator.py	251
ax/utils/testing/benchmark_stubs.py	116
ax/utils/testing/core_stubs.py	1164
ax/utils/testing/doctest.py	9
ax/utils/testing/fully_annotated.py	44
ax/utils/testing/manifest.py	72
ax/utils/testing/metrics/__init__.py	1
ax/utils/testing/metrics/backend_simulator_map.py	24
ax/utils/testing/metrics/branin_backend_map.py	42
ax/utils/testing/mock.py	64
ax/utils/testing/modeling_stubs.py	232
ax/utils/testing/pyre_strict.py	12
ax/utils/testing/torch_stubs.py	36
ax/utils/testing/unittest_conventions.py	31
ax/utils/tutorials/__init__.py	1
ax/utils/tutorials/cnn_utils.py	152
pyproject.toml	5
scripts/import_ax.py	5
scripts/insert_api_refs.py	46
scripts/make_tutorials.py	138
scripts/parse_sphinx.py	46
scripts/patch_site_config.py	41
scripts/update_versions_html.py	55
scripts/validate_sphinx.py	65
scripts/versions.js	91
scripts/wheels_build.ps1	28
setup.py	71
sphinx/source/conf.py	52
tutorials/Setup_and_Usage_of_BoTorch_Models_in_Ax.ipynb	1139
tutorials/botorch_modular.ipynb	627
tutorials/generation_strategy.ipynb	566
tutorials/modular_botax.ipynb	1257
tutorials/visualizations.ipynb	345
website/core/Footer.js	41
website/core/Tutorial.js	112
website/core/TutorialSidebar.js	77
website/pages/en/index.js	184
website/pages/tutorials/index.js	147
website/siteConfig.js	50
website/static/css/base_sphinx.css	500
website/static/css/custom.css	243
website/static/css/custom_sphinx.css	616
website/static/css/pygments.css	205
website/static/js/mathjax.js	15
website/static/js/plotUtils.js	42