bayinx 0.2.30__py3-none-any.whl → 0.2.32__py3-none-any.whl

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/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/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/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

{bayinx-0.2.30.dist-info → bayinx-0.2.32.dist-info}/METADATA

@@ -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.dist-info → bayinx-0.2.32.dist-info}/RECORD

@@ -1,9 +1,9 @@
 bayinx/__init__.py,sha256=l20JdkSsE_XGZlZFNEtySXf4NIlbjrao14vXPB-H6aQ,45
 bayinx/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bayinx/constraints/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bayinx/constraints/lower.py,sha256=O37qJ6ojRKbKGJlnQ7Vv7P2VGARcnKrOifwyUNBHma8,912
+bayinx/constraints/lower.py,sha256=MAAsWpZhqu1ySMskQ0fPVkCzW6CVDCU67q2bkCyzbLc,936
 bayinx/core/__init__.py,sha256=7vW2F8t3K4TWlSu5nZrYCdUrz5N9FMIfQQBn3IoeH6o,150
-bayinx/core/constraint.py,sha256=T1QP3WV9RU1nPW-9H7loc6DyBzElHVyDp1c4mmZpnmI,733
+bayinx/core/constraint.py,sha256=60KzDILVLQTCY3jt4YEnNKJ5gnG8IHTv_nNqrdwt_3c,751
 bayinx/core/flow.py,sha256=A5Vw5t76LPasnMgghjw6ulBkIm5L2jBprusVt-tuwko,2296
 bayinx/core/model.py,sha256=vfEnqBpHE2MtuJPIDgKvVYIv5n53E2e-KAAXEtqEy0c,2126
 bayinx/core/variational.py,sha256=2stsYKZDri1rLP7mrz7X2GWehBXNESdlWtmF2N9CEas,4787
@@ -17,13 +17,13 @@ bayinx/dists/censored/gamma2/r.py,sha256=3brRCKhE-74mRXyIyPcnyaWY2OJv8CZyUWPP9T1
 bayinx/mhx/__init__.py,sha256=AbpHGcgLb-kRsJGnwFEktk7uzpZOCcBY74-YBdrKVGs,1
 bayinx/mhx/vi/__init__.py,sha256=YfkXKsqo9Dk_AmQGjZKm4vfG8eLer2ez92G-cOExphs,193
 bayinx/mhx/vi/meanfield.py,sha256=BobfTagVGA5R-dclv-E0jSA80KZg1X6GGjiw7XR61vE,3643
-bayinx/mhx/vi/normalizing_flow.py,sha256=FvxDtqGRtaEeeF-bXCYnIEAvOOXVHKUK0oCTF9ma02Y,4622
+bayinx/mhx/vi/normalizing_flow.py,sha256=DYhvTiu2Fr5x8KpWAMZVUaio7ctG2X2SMUO0l8zfZ5g,4622
 bayinx/mhx/vi/standard.py,sha256=HaJsIz70Qo1Ql2hMQ-GQhcnfWiOGtyxgkOsm_yQaDKI,1718
 bayinx/mhx/vi/flows/__init__.py,sha256=Hn0Wqvvyv8Vr-mFmimwgNKCByxj-fjrlIvdR7tUSolg,180
-bayinx/mhx/vi/flows/fullaffine.py,sha256=s-fxgzv84BEqNMnmLt6vtvwJqBzixCC2OwWXLz3IK-w,1940
+bayinx/mhx/vi/flows/fullaffine.py,sha256=Z_G2Cg90Asgvqel8buMx5kEVsiIxDDh8rc-L_nP9OCY,1950
 bayinx/mhx/vi/flows/planar.py,sha256=WVj-oxcRctuoRA6KJjU63ek1ZgKNG2vI-TLN0QqjtKA,1916
 bayinx/mhx/vi/flows/radial.py,sha256=Obj3SraliawIHmP14F9wRpWt34y3kscY--Izy24eCvM,2499
 bayinx/mhx/vi/flows/sylvester.py,sha256=ppK0BmS_ThvrCEhJiP_-p-kj67TQHSlU_RUZpDbIhsQ,469
-bayinx-0.2.30.dist-info/METADATA,sha256=VVDldLMllMzTkx_tphb6k2n4u-PSwBbmFOGN-PG1BWc,3058
-bayinx-0.2.30.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-bayinx-0.2.30.dist-info/RECORD,,
+bayinx-0.2.32.dist-info/METADATA,sha256=C6xrmvyGJ573nlEObTgIb_CWiPDOY0zAtQj75qbgCc4,3058
+bayinx-0.2.32.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+bayinx-0.2.32.dist-info/RECORD,,