diff --git a/ax/models/torch/botorch_modular/utils.py b/ax/models/torch/botorch_modular/utils.py index c793d1aa4e4..2223a1467f1 100644 --- a/ax/models/torch/botorch_modular/utils.py +++ b/ax/models/torch/botorch_modular/utils.py @@ -303,7 +303,7 @@ def fit_botorch_model( fit_fully_bayesian_model_nuts( m, disable_progbar=True, - jit_compile=mll_options.get("jit_compile", False), + **mll_options, ) elif isinstance(m, (GPyTorchModel, PairwiseGP)): mll_options = mll_options or {}