Explain a parameter option and update blackjax dependency

Author: pnkraemer

probdiffeq/probdiffeq.py

@@ -260,6 +260,10 @@ def make_model(s):
             return ssm.conditional.to_derivative(tcoeff_index, s)
 
         model = func.vmap(make_model)(std)
+
+        # Use solve_triu=lstsq because for noise-free observations, the initial state
+        # of the ODE solution tends to be noise-free,
+        # which clashes and returns NaNs if we use exact solvers.
         return posterior.evaluate_lml(
             u, model=model, ssm=ssm, average_pdfs=average_pdfs, solve_triu=solve_triu
         )

pyproject.toml

@@ -48,7 +48,7 @@ doc = [
     "tueplots",
     "tqdm",
     "optax",
-    "blackjax>=1.0.0",
+    "blackjax[progress]>=1.0.0",
     "diffrax",
     "numba",
     # mkdocs 2.0 is a backwards-incompatible rewrite
@@ -141,7 +141,7 @@ ignore = [
     "D105",
     # Some backend's names shadow builtins, eg backend.abc. Ignore the warnings.
     "A005",
-    # Line-too-long is handled by the formatter, not the linter 
+    # Line-too-long is handled by the formatter, not the linter
     "E501",
 ]