Index

A

ACQF_KWARGS (ax.utils.common.constants.Keys attribute) aggregation() (ax.utils.sensitivity.derivative_measures.GpDGSMGpMean method) (ax.utils.sensitivity.derivative_measures.GpDGSMGpSampling method) agresti_coull_sem() (in module ax.utils.stats.statstools) all_trials (ax.utils.testing.backend_simulator.BackendSimulator property) allowed_to_fail() (in module ax.utils.common.executils) array_with_string_indices_to_dict() (in module ax.utils.sensitivity.sobol_measures) assert_is_instance_dict() (in module ax.utils.common.typeutils) assert_is_instance_list() (in module ax.utils.common.typeutils) assert_is_instance_of_tuple() (in module ax.utils.common.typeutils) assert_is_instance_optional() (in module ax.utils.common.typeutils) assertAllClose() (ax.utils.common.testutils.TestCase method) assertAxBaseEqual() (ax.utils.common.testutils.TestCase method) assertDictsAlmostEqual() (ax.utils.common.testutils.TestCase method) assertEqual() (ax.utils.common.testutils.TestCase method) assertIsSubDict() (ax.utils.common.testutils.TestCase method) assertRaisesOn() (ax.utils.common.testutils.TestCase method) Aug_Branin (class in ax.utils.measurement.synthetic_functions) Aug_Hartmann6 (class in ax.utils.measurement.synthetic_functions) ax.utils module ax.utils.common.base module ax.utils.common.constants module ax.utils.common.decorator module ax.utils.common.docutils module ax.utils.common.equality module ax.utils.common.executils module ax.utils.common.func_enum module ax.utils.common.kwargs module ax.utils.common.logger module ax.utils.common.mock module

ax.utils.common.random module ax.utils.common.result module ax.utils.common.serialization module ax.utils.common.testutils module ax.utils.common.timeutils module ax.utils.common.typeutils module ax.utils.common.typeutils_nonnative module ax.utils.common.typeutils_torch module ax.utils.measurement.synthetic_functions module ax.utils.sensitivity.derivative_gp module ax.utils.sensitivity.derivative_measures module ax.utils.sensitivity.sobol_measures module ax.utils.stats.model_fit_stats module ax.utils.stats.statstools module ax.utils.testing.backend_simulator module ax.utils.testing.mock module ax.utils.testing.test_init_files module ax.utils.testing.torch_stubs module ax.utils.testing.utils_testing_stubs module AX_ACQUISITION_KWARGS (ax.utils.common.constants.Keys attribute) ax_parameter_sens() (in module ax.utils.sensitivity.sobol_measures) AxOutputNameFilter (class in ax.utils.common.logger)

B

BackendSimulator (class in ax.utils.testing.backend_simulator) BackendSimulatorOptions (class in ax.utils.testing.backend_simulator) BackendSimulatorState (class in ax.utils.testing.backend_simulator) Base (class in ax.utils.common.base)

BATCH_INIT_CONDITIONS (ax.utils.common.constants.Keys attribute) bootstrap_indices (ax.utils.sensitivity.derivative_measures.GpDGSMGpMean attribute) Branin (class in ax.utils.measurement.synthetic_functions) build_file_handler() (in module ax.utils.common.logger) build_stream_handler() (in module ax.utils.common.logger)

C

callable_from_reference() (in module ax.utils.common.serialization) callable_to_reference() (in module ax.utils.common.serialization) CANDIDATE_SET (ax.utils.common.constants.Keys attribute) CANDIDATE_SIZE (ax.utils.common.constants.Keys attribute) ClassDecorator (class in ax.utils.common.decorator) coefficient_of_determination() (in module ax.utils.stats.model_fit_stats) completed (ax.utils.testing.backend_simulator.BackendSimulatorState attribute) (ax.utils.testing.backend_simulator.SimStatus attribute), [1]

compute_derivatives_from_model_list() (in module ax.utils.sensitivity.derivative_measures) compute_model_fit_metrics() (in module ax.utils.stats.model_fit_stats) compute_sobol_indices_from_model_list() (in module ax.utils.sensitivity.sobol_measures) consolidate_kwargs() (in module ax.utils.common.kwargs) copy_doc() (in module ax.utils.common.docutils) COST_AWARE_UTILITY (ax.utils.common.constants.Keys attribute) COST_INTERCEPT (ax.utils.common.constants.Keys attribute) current_timestamp_in_millis() (in module ax.utils.common.timeutils) CURRENT_VALUE (ax.utils.common.constants.Keys attribute)

D

dataframe_equals() (in module ax.utils.common.equality) datetime_equals() (in module ax.utils.common.equality) db_id (ax.utils.common.base.Base property) decorate_callable() (ax.utils.common.decorator.ClassDecorator method) (ax.utils.common.logger.disable_logger method) (ax.utils.common.logger.disable_loggers method) decorate_class() (ax.utils.common.decorator.ClassDecorator method)

DECORATE_PRIVATE (ax.utils.common.decorator.ClassDecorator attribute) DEFAULT_OBJECTIVE_NAME (ax.utils.common.constants.Keys attribute) deserialize_init_args() (ax.utils.common.serialization.SerializationMixin class method) dim (ax.utils.sensitivity.sobol_measures.SobolSensitivityGPSampling property) disable_logger (class in ax.utils.common.logger) disable_loggers (class in ax.utils.common.logger) domain (ax.utils.measurement.synthetic_functions.SyntheticFunction property)

E

entropy_of_observations() (in module ax.utils.stats.model_fit_stats) equality_typechecker() (in module ax.utils.common.equality) err (ax.utils.common.result.Err property) (ax.utils.common.result.Ok property) (ax.utils.common.result.Result property) Err (class in ax.utils.common.result) evalute_function() (ax.utils.sensitivity.sobol_measures.SobolSensitivity method)

exception (ax.utils.common.result.ExceptionE attribute) ExceptionE (class in ax.utils.common.result) execute_with_timeout() (in module ax.utils.common.executils) EXPAND (ax.utils.common.constants.Keys attribute) EXPECTED_ACQF_VAL (ax.utils.common.constants.Keys attribute) EXPERIMENT_TOTAL_CONCURRENT_ARMS (ax.utils.common.constants.Keys attribute) extract_init_args() (in module ax.utils.common.serialization)

F

f() (ax.utils.measurement.synthetic_functions.SyntheticFunction method) FACTORIAL_PLUS_EMPIRICAL_BAYES_THOMPSON_SAMPLING (ax.utils.common.constants.Keys attribute) failed (ax.utils.testing.backend_simulator.BackendSimulatorState attribute) (ax.utils.testing.backend_simulator.SimStatus attribute), [1] failure_rate (ax.utils.testing.backend_simulator.BackendSimulator property) (ax.utils.testing.backend_simulator.BackendSimulatorOptions attribute) FIDELITY_FEATURES (ax.utils.common.constants.Keys attribute) FIDELITY_WEIGHTS (ax.utils.common.constants.Keys attribute) filter() (ax.utils.common.logger.AxOutputNameFilter method) filter_kwargs() (in module ax.utils.common.kwargs)

first_order_indices() (ax.utils.sensitivity.sobol_measures.SobolSensitivity method) (ax.utils.sensitivity.sobol_measures.SobolSensitivityGPMean method) (ax.utils.sensitivity.sobol_measures.SobolSensitivityGPSampling method) fmax (ax.utils.measurement.synthetic_functions.SyntheticFunction property) fmin (ax.utils.measurement.synthetic_functions.SyntheticFunction property) format() (in module ax.utils.testing.backend_simulator) FRAC_RANDOM (ax.utils.common.constants.Keys attribute) from_botorch() (in module ax.utils.measurement.synthetic_functions) FromBotorch (class in ax.utils.measurement.synthetic_functions) FULL_PARAMETERIZATION (ax.utils.common.constants.Keys attribute) FuncEnum (class in ax.utils.common.func_enum)

G

GaussianLinkMean() (in module ax.utils.sensitivity.sobol_measures) generate_all_input_matrix() (ax.utils.sensitivity.sobol_measures.SobolSensitivity method) get_backend_simulator_with_trials() (in module ax.utils.testing.utils_testing_stubs) get_function_argument_names() (in module ax.utils.common.kwargs) get_function_default_arguments() (in module ax.utils.common.kwargs) get_KxX_dx() (in module ax.utils.sensitivity.derivative_gp) get_Kxx_dx2() (in module ax.utils.sensitivity.derivative_gp) get_KXX_inv() (in module ax.utils.sensitivity.derivative_gp)

get_logger() (in module ax.utils.common.logger) get_sim_trial_by_index() (ax.utils.testing.backend_simulator.BackendSimulator method) get_torch_test_data() (in module ax.utils.testing.torch_stubs) GpDGSMGpMean (class in ax.utils.sensitivity.derivative_measures) GpDGSMGpSampling (class in ax.utils.sensitivity.derivative_measures) gradient_absolute_measure() (ax.utils.sensitivity.derivative_measures.GpDGSMGpMean method) gradient_measure() (ax.utils.sensitivity.derivative_measures.GpDGSMGpMean method) gradients_square_measure() (ax.utils.sensitivity.derivative_measures.GpDGSMGpMean method)

H

handle_exceptions_in_retries() (in module ax.utils.common.executils)

Hartmann6 (class in ax.utils.measurement.synthetic_functions)

I

IMMUTABLE_SEARCH_SPACE_AND_OPT_CONF (ax.utils.common.constants.Keys attribute) informative_failure_on_none() (ax.utils.measurement.synthetic_functions.SyntheticFunction method) InitTest (class in ax.utils.testing.test_init_files) internal_clock (ax.utils.testing.backend_simulator.BackendSimulatorOptions attribute) inverse_variance_weight() (in module ax.utils.stats.statstools) is_ax_equal() (in module ax.utils.common.equality)

is_err() (ax.utils.common.result.Err method) (ax.utils.common.result.Ok method) (ax.utils.common.result.Result method) is_ok() (ax.utils.common.result.Err method) (ax.utils.common.result.Ok method) (ax.utils.common.result.Result method)

K

Keys (class in ax.utils.common.constants)

L

log() (ax.utils.testing.backend_simulator.BackendSimulator method)

LONG_RUN (ax.utils.common.constants.Keys attribute) lookup_trial_index_status() (ax.utils.testing.backend_simulator.BackendSimulator method)

M

make_indices_str() (in module ax.utils.common.logger) map() (ax.utils.common.result.Err method) (ax.utils.common.result.Ok method) (ax.utils.common.result.Result method) map_err() (ax.utils.common.result.Err method) (ax.utils.common.result.Ok method) (ax.utils.common.result.Result method) map_or() (ax.utils.common.result.Err method) (ax.utils.common.result.Ok method) (ax.utils.common.result.Result method) map_or_else() (ax.utils.common.result.Err method) (ax.utils.common.result.Ok method) (ax.utils.common.result.Result method) marginal_effects() (in module ax.utils.stats.statstools) max_concurrency (ax.utils.testing.backend_simulator.BackendSimulator property) (ax.utils.testing.backend_simulator.BackendSimulatorOptions attribute) MAXIMIZE (ax.utils.common.constants.Keys attribute) (ax.utils.stats.model_fit_stats.ModelFitMetricDirection attribute) maximums (ax.utils.measurement.synthetic_functions.SyntheticFunction property) mean_gradients (ax.utils.sensitivity.derivative_measures.GpDGSMGpMean attribute) mean_gradients_btsp (ax.utils.sensitivity.derivative_measures.GpDGSMGpMean attribute) mean_of_the_standardized_error() (in module ax.utils.stats.model_fit_stats) message (ax.utils.common.result.ExceptionE attribute) METADATA (ax.utils.common.constants.Keys attribute) METRIC_NAMES (ax.utils.common.constants.Keys attribute) minimal_optimize_with_nsgaii() (in module ax.utils.testing.mock) MINIMIZE (ax.utils.stats.model_fit_stats.ModelFitMetricDirection attribute) minimums (ax.utils.measurement.synthetic_functions.SyntheticFunction property) mock_botorch_optimize() (in module ax.utils.testing.mock) mock_botorch_optimize_context_manager() (in module ax.utils.testing.mock) mock_patch_method_original() (in module ax.utils.common.mock) ModelFitMetricDirection (class in ax.utils.stats.model_fit_stats)

ModelFitMetricProtocol (class in ax.utils.stats.model_fit_stats) module ax.utils 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.func_enum 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.measurement.synthetic_functions 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_simulator ax.utils.testing.mock ax.utils.testing.test_init_files ax.utils.testing.torch_stubs ax.utils.testing.utils_testing_stubs

N

name (ax.utils.measurement.synthetic_functions.FromBotorch property) (ax.utils.measurement.synthetic_functions.SyntheticFunction property) named_tuple_to_dict() (in module ax.utils.common.serialization) new_trial() (ax.utils.testing.backend_simulator.BackendSimulator method) num_completed (ax.utils.testing.backend_simulator.BackendSimulator property) num_failed (ax.utils.testing.backend_simulator.BackendSimulator property)

NUM_FANTASIES (ax.utils.common.constants.Keys attribute) NUM_INNER_RESTARTS (ax.utils.common.constants.Keys attribute) num_queued (ax.utils.testing.backend_simulator.BackendSimulator property) NUM_RESTARTS (ax.utils.common.constants.Keys attribute) num_running (ax.utils.testing.backend_simulator.BackendSimulator property) NUM_TRACE_OBSERVATIONS (ax.utils.common.constants.Keys attribute) numpy_type_to_python_type() (in module ax.utils.common.typeutils_nonnative)

O

object_attribute_dicts_equal() (in module ax.utils.common.equality) object_attribute_dicts_find_unequal_fields() (in module ax.utils.common.equality) ok (ax.utils.common.result.Err property) (ax.utils.common.result.Ok property) (ax.utils.common.result.Result property)

Ok (class in ax.utils.common.result) OPTIMIZER_KWARGS (ax.utils.common.constants.Keys attribute) options (ax.utils.testing.backend_simulator.BackendSimulatorState attribute)

P

PAIRWISE_PREFERENCE_QUERY (ax.utils.common.constants.Keys attribute) positive_part_james_stein() (in module ax.utils.stats.statstools) posterior_derivative() (in module ax.utils.sensitivity.derivative_gp)

PREFERENCE_DATA (ax.utils.common.constants.Keys attribute) ProbitLinkMean() (in module ax.utils.sensitivity.sobol_measures) PROJECT (ax.utils.common.constants.Keys attribute)

Q

QMC (ax.utils.common.constants.Keys attribute)

queued (ax.utils.testing.backend_simulator.BackendSimulatorState attribute) (ax.utils.testing.backend_simulator.SimStatus attribute), [1]

R

RAW_INNER_SAMPLES (ax.utils.common.constants.Keys attribute) RAW_SAMPLES (ax.utils.common.constants.Keys attribute) required_dimensionality (ax.utils.measurement.synthetic_functions.SyntheticFunction property) Result (class in ax.utils.common.result)

RESUMED_FROM_STORAGE_TS (ax.utils.common.constants.Keys attribute) retry_on_exception() (in module ax.utils.common.executils) run_trial() (ax.utils.testing.backend_simulator.BackendSimulator method) running (ax.utils.testing.backend_simulator.BackendSimulatorState attribute) (ax.utils.testing.backend_simulator.SimStatus attribute), [1]

S

same_elements() (in module ax.utils.common.equality) sample_discrete_parameters() (in module ax.utils.sensitivity.derivative_measures) SAMPLER (ax.utils.common.constants.Keys attribute) samples_gradients (ax.utils.sensitivity.derivative_measures.GpDGSMGpSampling attribute) samples_gradients_btsp (ax.utils.sensitivity.derivative_measures.GpDGSMGpSampling attribute) second_order_indices() (ax.utils.sensitivity.sobol_measures.SobolSensitivity method) (ax.utils.sensitivity.sobol_measures.SobolSensitivityGPMean method) (ax.utils.sensitivity.sobol_measures.SobolSensitivityGPSampling method) SEED_INNER (ax.utils.common.constants.Keys attribute) SEQUENTIAL (ax.utils.common.constants.Keys attribute) SerializationMixin (class in ax.utils.common.serialization) serialize_init_args() (ax.utils.common.serialization.SerializationMixin class method) (in module ax.utils.common.serialization) set_ax_logger_levels() (in module ax.utils.common.logger) set_rng_seed() (in module ax.utils.common.random) set_stderr_log_level() (in module ax.utils.common.logger) setUp() (ax.utils.common.testutils.TestCase method) SHORT_RUN (ax.utils.common.constants.Keys attribute) silence_stderr() (ax.utils.common.testutils.TestCase static method)

sim_completed_time (ax.utils.testing.backend_simulator.SimTrial attribute), [1] sim_queued_time (ax.utils.testing.backend_simulator.SimTrial attribute), [1] sim_runtime (ax.utils.testing.backend_simulator.SimTrial attribute), [1] sim_start_time (ax.utils.testing.backend_simulator.SimTrial attribute), [1] SimStatus (class in ax.utils.testing.backend_simulator) SimTrial (class in ax.utils.testing.backend_simulator) skip_fit_gpytorch_mll() (in module ax.utils.testing.mock) skip_fit_gpytorch_mll_context_manager() (in module ax.utils.testing.mock) SobolSensitivity (class in ax.utils.sensitivity.sobol_measures) SobolSensitivityGPMean (class in ax.utils.sensitivity.sobol_measures) SobolSensitivityGPSampling (class in ax.utils.sensitivity.sobol_measures) SortableBase (class in ax.utils.common.base) state() (ax.utils.testing.backend_simulator.BackendSimulator method) STATE_DICT (ax.utils.common.constants.Keys attribute) status() (ax.utils.testing.backend_simulator.BackendSimulator method) std_of_the_standardized_error() (in module ax.utils.stats.model_fit_stats) stop_trial() (ax.utils.testing.backend_simulator.BackendSimulator method) SUBCLASS (ax.utils.common.constants.Keys attribute) SUBSET_MODEL (ax.utils.common.constants.Keys attribute) SyntheticFunction (class in ax.utils.measurement.synthetic_functions)

T

TASK_FEATURE_NAME (ax.utils.common.constants.Keys attribute) TASK_FEATURES (ax.utils.common.constants.Keys attribute) tb_str() (ax.utils.common.result.ExceptionE method) test_InitFiles() (ax.utils.testing.test_init_files.InitTest method) TestCase (class in ax.utils.common.testutils) time (ax.utils.testing.backend_simulator.BackendSimulator property) time_remaining (ax.utils.testing.backend_simulator.SimStatus attribute), [1] time_scaling (ax.utils.testing.backend_simulator.BackendSimulator property) (ax.utils.testing.backend_simulator.BackendSimulatorOptions attribute) timestamps_in_range() (in module ax.utils.common.timeutils)

to_ds() (in module ax.utils.common.timeutils) to_ts() (in module ax.utils.common.timeutils) torch_type_from_str() (in module ax.utils.common.typeutils_torch) torch_type_to_str() (in module ax.utils.common.typeutils_torch) total_order_indices() (ax.utils.sensitivity.sobol_measures.SobolSensitivity method) (ax.utils.sensitivity.sobol_measures.SobolSensitivityGPMean method) (ax.utils.sensitivity.sobol_measures.SobolSensitivityGPSampling method) total_variance() (in module ax.utils.stats.statstools) TRIAL_COMPLETION_TIMESTAMP (ax.utils.common.constants.Keys attribute) trial_index (ax.utils.testing.backend_simulator.SimTrial attribute), [1]

U

unixtime_to_pandas_ts() (in module ax.utils.common.timeutils) UNKNOWN_GENERATION_NODE (ax.utils.common.constants.Keys attribute) UNNAMED_ARM (ax.utils.common.constants.Keys attribute) unwrap() (ax.utils.common.result.Err method) (ax.utils.common.result.Ok method) (ax.utils.common.result.Result method) unwrap_err() (ax.utils.common.result.Err method) (ax.utils.common.result.Ok method) (ax.utils.common.result.Result method)

unwrap_or() (ax.utils.common.result.Err method) (ax.utils.common.result.Ok method) (ax.utils.common.result.Result method) unwrap_or_else() (ax.utils.common.result.Err method) (ax.utils.common.result.Ok method) (ax.utils.common.result.Result method) UnwrapError update() (ax.utils.testing.backend_simulator.BackendSimulator method) use_internal_clock (ax.utils.testing.backend_simulator.BackendSimulator property) use_update_as_start_time (ax.utils.testing.backend_simulator.BackendSimulatorOptions attribute)

V

value (ax.utils.common.result.Err property) (ax.utils.common.result.Ok property) (ax.utils.common.result.Result property)

verbose_logging (ax.utils.testing.backend_simulator.BackendSimulatorState attribute)

W

WARM_START_REFITTING (ax.utils.common.constants.Keys attribute) WARMSTART_TRIAL_MODEL_KEY (ax.utils.common.constants.Keys attribute)

warn_on_kwargs() (in module ax.utils.common.kwargs) with_rng_seed() (in module ax.utils.common.random)

X

X_BASELINE (ax.utils.common.constants.Keys attribute)