All modules for which code is available

• abc
• ax.benchmark.benchmark
• ax.benchmark.benchmark_method
• ax.benchmark.benchmark_problem
• ax.benchmark.benchmark_result
• ax.benchmark.methods.modular_botorch
• ax.benchmark.methods.sobol
• ax.benchmark.metrics.base
• ax.benchmark.metrics.benchmark
• ax.benchmark.metrics.jenatton
• ax.benchmark.problems.hd_embedding
• ax.benchmark.problems.hpo.pytorch_cnn
• ax.benchmark.problems.hpo.torchvision
• ax.benchmark.problems.registry
• ax.benchmark.problems.surrogate
• ax.benchmark.problems.synthetic.discretized.mixed_integer
• ax.benchmark.problems.synthetic.hss.jenatton
• ax.benchmark.runners.base
• ax.benchmark.runners.botorch_test
• ax.benchmark.runners.surrogate
• ax.core.arm
• ax.core.base_trial
• ax.core.batch_trial
• ax.core.data
• ax.core.experiment
• ax.core.formatting_utils
• ax.core.generation_strategy_interface
• ax.core.generator_run
• ax.core.map_data
• ax.core.map_metric
• ax.core.metric
• ax.core.multi_type_experiment
• ax.core.objective
• ax.core.observation
• ax.core.optimization_config
• ax.core.outcome_constraint
• ax.core.parameter
• ax.core.parameter_constraint
• ax.core.parameter_distribution
• ax.core.risk_measures
• ax.core.runner
• ax.core.search_space
• ax.core.trial
• ax.core.types
• ax.core.utils
• ax.early_stopping.strategies.base
• ax.early_stopping.strategies.logical
• ax.early_stopping.strategies.percentile
• ax.early_stopping.strategies.threshold
• ax.early_stopping.utils
• ax.exceptions.core
• ax.exceptions.data_provider
• ax.exceptions.generation_strategy
• ax.exceptions.model
• ax.exceptions.storage
• ax.global_stopping.strategies.base
• ax.global_stopping.strategies.improvement
• ax.metrics.botorch_test_problem
• ax.metrics.branin
• ax.metrics.branin_map
• ax.metrics.chemistry
• ax.metrics.curve
• ax.metrics.dict_lookup
• ax.metrics.factorial
• ax.metrics.hartmann6
• ax.metrics.l2norm
• ax.metrics.noisy_function
• ax.metrics.noisy_function_map
• ax.metrics.sklearn
• ax.metrics.tensorboard
• ax.metrics.torchx
• ax.modelbridge.base
• ax.modelbridge.best_model_selector
• ax.modelbridge.cross_validation
• ax.modelbridge.discrete
• ax.modelbridge.dispatch_utils
• ax.modelbridge.external_generation_node
• ax.modelbridge.factory
• ax.modelbridge.generation_node
• ax.modelbridge.generation_strategy
• ax.modelbridge.map_torch
• ax.modelbridge.model_spec
• ax.modelbridge.modelbridge_utils
• ax.modelbridge.pairwise
• ax.modelbridge.prediction_utils
• ax.modelbridge.random
• ax.modelbridge.registry
• ax.modelbridge.torch
• ax.modelbridge.transforms.base
• ax.modelbridge.transforms.cap_parameter
• ax.modelbridge.transforms.cast
• ax.modelbridge.transforms.centered_unit_x
• ax.modelbridge.transforms.choice_encode
• ax.modelbridge.transforms.convert_metric_names
• ax.modelbridge.transforms.derelativize
• ax.modelbridge.transforms.int_range_to_choice
• ax.modelbridge.transforms.int_to_float
• ax.modelbridge.transforms.inverse_gaussian_cdf_y
• ax.modelbridge.transforms.ivw
• ax.modelbridge.transforms.log
• ax.modelbridge.transforms.log_y
• ax.modelbridge.transforms.logit
• ax.modelbridge.transforms.map_unit_x
• ax.modelbridge.transforms.merge_repeated_measurements
• ax.modelbridge.transforms.metrics_as_task
• ax.modelbridge.transforms.one_hot
• ax.modelbridge.transforms.percentile_y
• ax.modelbridge.transforms.power_transform_y
• ax.modelbridge.transforms.relativize
• ax.modelbridge.transforms.remove_fixed
• ax.modelbridge.transforms.rounding
• ax.modelbridge.transforms.search_space_to_choice
• ax.modelbridge.transforms.standardize_y
• ax.modelbridge.transforms.stratified_standardize_y
• ax.modelbridge.transforms.task_encode
• ax.modelbridge.transforms.time_as_feature
• ax.modelbridge.transforms.trial_as_task
• ax.modelbridge.transforms.unit_x
• ax.modelbridge.transforms.utils
• ax.modelbridge.transforms.winsorize
• ax.models.base
• ax.models.discrete.eb_thompson
• ax.models.discrete.full_factorial
• ax.models.discrete.thompson
• ax.models.discrete_base
• ax.models.model_utils
• ax.models.random.base
• ax.models.random.sobol
• ax.models.random.uniform
• ax.models.torch.botorch
• ax.models.torch.botorch_defaults
• ax.models.torch.botorch_kg
• ax.models.torch.botorch_mes
• ax.models.torch.botorch_modular.acquisition
• ax.models.torch.botorch_modular.kernels
• ax.models.torch.botorch_modular.model
• ax.models.torch.botorch_modular.sebo
• ax.models.torch.botorch_modular.surrogate
• ax.models.torch.botorch_modular.utils
• ax.models.torch.botorch_moo
• ax.models.torch.botorch_moo_defaults
• ax.models.torch.cbo_lcea
• ax.models.torch.cbo_lcem
• ax.models.torch.cbo_sac
• ax.models.torch.posterior_mean
• ax.models.torch.randomforest
• ax.models.torch.utils
• ax.models.torch_base
• ax.models.winsorization_config
• ax.plot.bandit_rollout
• ax.plot.base
• ax.plot.color
• ax.plot.contour
• ax.plot.diagnostic
• ax.plot.feature_importances
• ax.plot.helper
• ax.plot.marginal_effects
• ax.plot.parallel_coordinates
• ax.plot.pareto_frontier
• ax.plot.pareto_utils
• ax.plot.render
• ax.plot.scatter
• ax.plot.slice
• ax.plot.table_view
• ax.plot.trace
• ax.runners.simulated_backend
• ax.runners.single_running_trial_mixin
• ax.runners.synthetic
• ax.runners.torchx
• ax.service.ax_client
• ax.service.interactive_loop
• ax.service.managed_loop
• ax.service.scheduler
• ax.service.utils.best_point
• ax.service.utils.best_point_mixin
• ax.service.utils.early_stopping
• ax.service.utils.instantiation
• ax.service.utils.report_utils
• ax.service.utils.scheduler_options
• ax.service.utils.with_db_settings_base
• ax.storage.botorch_modular_registry
• ax.storage.json_store.decoder
• ax.storage.json_store.decoders
• ax.storage.json_store.encoder
• ax.storage.json_store.encoders
• ax.storage.json_store.load
• ax.storage.json_store.save
• ax.storage.metric_registry
• ax.storage.registry_bundle
• ax.storage.runner_registry
• ax.storage.sqa_store.db
• ax.storage.sqa_store.decoder
• ax.storage.sqa_store.delete
• ax.storage.sqa_store.encoder
• ax.storage.sqa_store.json
• ax.storage.sqa_store.load
• ax.storage.sqa_store.reduced_state
• ax.storage.sqa_store.save
• ax.storage.sqa_store.sqa_classes
• ax.storage.sqa_store.sqa_config
• ax.storage.sqa_store.sqa_enum
• ax.storage.sqa_store.structs
• ax.storage.sqa_store.timestamp
• ax.storage.sqa_store.utils
• ax.storage.sqa_store.validation
• ax.storage.utils
• ax.telemetry.experiment
• ax.telemetry.generation_strategy
• ax.telemetry.scheduler
• ax.utils.common.base
• ax.utils.common.constants
• ax.utils.common.decorator
• ax.utils.common.docutils
• ax.utils.common.equality
• ax.utils.common.executils
• ax.utils.common.kwargs
• ax.utils.common.logger
• ax.utils.common.mock
• ax.utils.common.random
• ax.utils.common.result
• ax.utils.common.serialization
• ax.utils.common.testutils
• ax.utils.common.timeutils
• ax.utils.common.typeutils
• ax.utils.common.typeutils_nonnative
• ax.utils.common.typeutils_torch
• ax.utils.flake8_plugins.docstring_checker
• ax.utils.measurement.synthetic_functions
• ax.utils.notebook.plotting
• ax.utils.report.render
• ax.utils.sensitivity.derivative_gp
• ax.utils.sensitivity.derivative_measures
• ax.utils.sensitivity.sobol_measures
• ax.utils.stats.model_fit_stats
• ax.utils.stats.statstools
• ax.utils.testing.backend_scheduler
• ax.utils.testing.backend_simulator
• ax.utils.testing.benchmark_stubs
• ax.utils.testing.core_stubs
• ax.utils.testing.metrics.backend_simulator_map
• ax.utils.testing.metrics.branin_backend_map
• ax.utils.testing.mock
• ax.utils.testing.modeling_stubs
• ax.utils.testing.preference_stubs
• ax.utils.testing.test_init_files
• ax.utils.testing.torch_stubs
• ax.utils.testing.utils
• ax.utils.tutorials.cnn_utils
• sqlalchemy.ext.mutable.MutableDict
• sqlalchemy.ext.mutable.MutableList