Remove NaN detection from blackjax example because somewhere, blackjax uses NaNs

pnkraemer · pnkraemer · commit 9d9d36d91d0e · 2026-02-17T15:14:12.000+01:00
diff --git a/docs/examples_advanced/parameter_estimation_blackjax.py b/docs/examples_advanced/parameter_estimation_blackjax.py
@@ -134,10 +134,6 @@
 
 from probdiffeq import ivpsolve, probdiffeq, taylor
 
-# Fail this notebook on NaN detection (to catch those in the CI)
-jax.config.update("jax_debug_nans", True)
-
-
 # +
 
 # IVP examples in JAX
