Update jax.py

Author: heptaflar

uncertaintyx/m/jax.py

@@ -137,7 +137,7 @@ def f(p, x):
     def estimate(
         self, x: np.ndarray | None = None, y: np.ndarray | None = None
     ) -> np.ndarray:
-        return np.array([1.0, 0.0])
+        return np.array([1.0, 1.0, 0.0])
 
 
 class Linear(ToM):