bayinx 0.2.30__tar.gz → 0.2.32__tar.gz

{bayinx-0.2.30 → bayinx-0.2.32}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: bayinx
-Version: 0.2.30
+Version: 0.2.32
 Summary: Bayesian Inference with JAX
 Requires-Python: >=3.12
 Requires-Dist: equinox>=0.11.12

{bayinx-0.2.30 → bayinx-0.2.32}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "bayinx"
-version = "0.2.30"
+version = "0.2.32"
 description = "Bayesian Inference with JAX"
 readme = "README.md"
 requires-python = ">=3.12"
@@ -19,7 +19,7 @@ build-backend = "hatchling.build"
 addopts = "-q --benchmark-min-rounds=30 --benchmark-columns=rounds,mean,median,stddev --benchmark-group-by=func"
 
 [tool.bumpversion]
-current_version = "0.2.30"
+current_version = "0.2.32"
 parse = "(?P<major>\\d+)\\.(?P<minor>\\d+)\\.(?P<patch>\\d+)"
 serialize = ["{major}.{minor}.{patch}"]
 search = "{current_version}"

{bayinx-0.2.30 → bayinx-0.2.32}/src/bayinx/constraints/lower.py

@@ -6,7 +6,7 @@ from jaxtyping import Array, ArrayLike, Scalar, ScalarLike
 from bayinx.core.constraint import Constraint
 
 
-class LowerBound(Constraint):
+class Lower(Constraint):
     """
     Enforces a lower bound on the parameter.
     """
@@ -18,7 +18,7 @@ class LowerBound(Constraint):
 
     def constrain(self, x: ArrayLike) -> Tuple[Array, Scalar]:
         """
-        Applies the lower bound constraint and computes the laj.
+        Applies the lower bound constraint and adjusts the posterior density.
 
         # Parameters
         - `x`: The unconstrained JAX Array-like input.
@@ -26,12 +26,12 @@ class LowerBound(Constraint):
         # Parameters
         A tuple containing:
             - The constrained JAX Array (x > self.lb).
-            - A scalar JAX Array representing the laj of the transformation.
+            - A scalar JAX Array representing the log-absolute-Jacobian of the transformation.
         """
         # Compute transformation adjustment
-        ladj: Scalar = jnp.sum(x)
+        laj: Scalar = jnp.sum(x)
 
         # Compute transformation
         x = jnp.exp(x) + self.lb
 
-        return x, ladj
+        return x, laj

{bayinx-0.2.30 → bayinx-0.2.32}/src/bayinx/core/constraint.py

@@ -13,7 +13,7 @@ class Constraint(eqx.Module):
     @abstractmethod
     def constrain(self, x: ArrayLike) -> Tuple[Array, Scalar]:
         """
-        Applies the constraining transformation to an unconstrained input and computes the log-absolute-jacobian of the transformation.
+        Applies the constraining transformation to an unconstrained input and computes the log-absolute-Jacobian of the transformation.
 
         # Parameters
         - `x`: The unconstrained JAX Array-like input.
@@ -21,6 +21,6 @@ class Constraint(eqx.Module):
         # Returns
         A tuple containing:
             - The constrained JAX Array.
-            - A scalar JAX Array representing the laj of the transformation.
+            - A scalar JAX Array representing the log-absolute-Jacobian of the transformation.
         """
         pass

{bayinx-0.2.30 → bayinx-0.2.32}/src/bayinx/mhx/vi/flows/fullaffine.py

@@ -40,7 +40,7 @@ class FullAffine(Flow):
                 diag: Array = jnp.exp(jnp.diag(scale))
 
                 # Return matrix with modified diagonal
-                return jnp.fill_diagonal(scale, diag, inplace=False)
+                return jnp.fill_diagonal(jnp.tril(scale), diag, inplace=False)
 
             self.constraints = {"scale": constrain_scale}
 

{bayinx-0.2.30 → bayinx-0.2.32}/src/bayinx/mhx/vi/normalizing_flow.py

@@ -75,7 +75,7 @@ class NormalizingFlow(Variational):
 
         for map in self.flows:
             # Compute adjustment
-            laj, draws = map.adjust_density(draws)
+            draws, laj = map.adjust_density(draws)
 
             # Adjust variational density
             variational_evals = variational_evals - laj