BUG: pm.sample 无法禁用外部 JAX 采样器 (numpyro/blackjax) 的初始值抖动
作者: AlexanderFengler创建于 2026年7月9日更新于 2026年8月24日
标签bug
描述问题: pymc.sampling.jax.sample_jax_nuts 有一个 jitter: bool = True 参数,但通过 pm.sample(nuts_sampler="numpyro"|"blackjax") 无法访问它。_sample_external_nuts 调用 sample_jax_nuts(...) 而不传递 jitter,因此它始终为 True,通过 nuts= / nuts_sampler_kwargs 传递它会进入 NUTS 内核:
内容来源: pymc-devs/pymc