bayinx 0.2.21__py3-none-any.whl → 0.2.22__py3-none-any.whl

bayinx/dists/normal.py

@@ -1,7 +1,7 @@
 import jax.lax as _lax
 from jaxtyping import Array, ArrayLike, Float, Real
 
-_PI = 3.141592653589793
+__PI = 3.141592653589793
 
 
 def prob(
@@ -20,7 +20,7 @@ def prob(
     """
 
     return _lax.exp(-0.5 * _lax.square((x - mu) / sigma)) / ( # pyright: ignore
-        sigma * _lax.sqrt(2.0 * _PI)
+        sigma * _lax.sqrt(2.0 * __PI)
     )
 
 
@@ -39,7 +39,7 @@ def logprob(
     The log of the PDF evaluated at `x`. The output will have the broadcasted shapes of `x`, `mu`, and `sigma`.
     """
 
-    return -_lax.log(sigma * _lax.sqrt(2.0 * _PI)) - 0.5 * _lax.square((x - mu) / sigma) # pyright: ignore
+    return -_lax.log(sigma * _lax.sqrt(2.0 * __PI)) - 0.5 * _lax.square((x - mu) / sigma) # pyright: ignore
 
 
 def uprob(

bayinx/dists/uniform.py

@@ -0,0 +1,75 @@
+import jax.lax as _lax
+import jax.numpy as jnp
+from jaxtyping import Array, ArrayLike, Float, Real
+
+
+def prob(
+    x: Real[ArrayLike, "..."], lb: Real[ArrayLike, "..."], ub: Real[ArrayLike, "..."]
+) -> Float[Array, "..."]:
+    """
+    The probability density function (PDF) for a Uniform distribution.
+
+    # Parameters
+    - `x`:      Value(s) at which to evaluate the PDF.
+    - `lb`:     The lower bound parameter(s).
+    - `ub`:     The upper bound parameter(s).
+
+    # Returns
+    The PDF evaluated at `x`. The output will have the broadcasted shapes of `x`, `lb`, and `ub`.
+    """
+
+    return 1.0 / (ub - lb) # pyright: ignore
+
+
+def logprob(
+    x: Real[ArrayLike, "..."], lb: Real[ArrayLike, "..."], ub: Real[ArrayLike, "..."]
+) -> Float[Array, "..."]:
+    """
+    The log of the probability density function (log PDF) for a Uniform distribution.
+
+    # Parameters
+    - `x`:      Value(s) at which to evaluate the PDF.
+    - `lb`:     The lower bound parameter(s).
+    - `ub`:     The upper bound parameter(s).
+
+    # Returns
+    The log of the PDF evaluated at `x`. The output will have the broadcasted shapes of `x`, `lb`, and `ub`.
+    """
+
+    return _lax.log(1.0) - _lax.log(ub - lb) # pyright: ignore
+
+
+def uprob(
+    x: Real[ArrayLike, "..."], lb: Real[ArrayLike, "..."], ub: Real[ArrayLike, "..."]
+) -> Float[Array, "..."]:
+    """
+    The unnormalized probability density function (uPDF) for a Uniform distribution.
+
+    # Parameters
+    - `x`:      Value(s) at which to evaluate the PDF.
+    - `lb`:     The lower bound parameter(s).
+    - `ub`:     The upper bound parameter(s).
+
+    # Returns
+    The uPDF evaluated at `x`. The output will have the broadcasted shapes of `x`, `lb`, and `ub`.
+    """
+
+    return jnp.ones(jnp.broadcast_arrays(x,lb,ub))
+
+
+def ulogprob(
+    x: Real[ArrayLike, "..."], lb: Real[ArrayLike, "..."], ub: Real[ArrayLike, "..."]
+) -> Float[Array, "..."]:
+    """
+    The log of the unnormalized probability density function (log uPDF) for a Uniform distribution.
+
+    # Parameters
+    - `x`:      Value(s) at which to evaluate the PDF.
+    - `lb`:     The lower bound parameter(s).
+    - `ub`:     The upper bound parameter(s).
+
+    # Returns
+    The log uPDF evaluated at `x`. The output will have the broadcasted shapes of `x`, `lb`, and `ub`.
+    """
+
+    return jnp.zeros(jnp.broadcast_arrays(x,lb,ub))

{bayinx-0.2.21.dist-info → bayinx-0.2.22.dist-info}/METADATA

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

{bayinx-0.2.21.dist-info → bayinx-0.2.22.dist-info}/RECORD

@@ -10,7 +10,8 @@ bayinx/dists/bernoulli.py,sha256=xMV9BgtVX_1XkPdZ43q0meMIEkgMyuUPx--dyo6_DKs,100
 bayinx/dists/binomial.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bayinx/dists/gamma.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 bayinx/dists/gamma2.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-bayinx/dists/normal.py,sha256=cfstoFSXeX3RS9AUyjKyjRR4G2Mi36o2FhyT1cfRTNw,2659
+bayinx/dists/normal.py,sha256=rtSDi0NAObH1LGRWiPZk_6cbSVv2dOPHkgxtWn6gFgM,2662
+bayinx/dists/uniform.py,sha256=PSZIIc2QfNF5XYi-TLGltnr_vnAIA-MZsn1rKV8QXAo,2353
 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=LNLwfjKO9os7YBmRBpGEiFxlxonuN7DHVFEmXV3hvfI,3876
@@ -21,6 +22,6 @@ bayinx/mhx/vi/flows/fullaffine.py,sha256=2QbOtA1Jmu-yRcJeFmCKc8N1atm8G7JXYMLEZaE
 bayinx/mhx/vi/flows/planar.py,sha256=qmtWpIBXRct2seI78pkmtF0X7cASUBELqmZmf2QS5Gs,1918
 bayinx/mhx/vi/flows/radial.py,sha256=c-SWybGn_jmgBQk9ZMQ5uHKPzcdhowNp8MD8t1-8VZM,2501
 bayinx/mhx/vi/flows/sylvester.py,sha256=ppK0BmS_ThvrCEhJiP_-p-kj67TQHSlU_RUZpDbIhsQ,469
-bayinx-0.2.21.dist-info/METADATA,sha256=0V_u_U7-AHlu97D6S6P9aU6iQPDvvcseOMk9W0f-iP4,3058
-bayinx-0.2.21.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-bayinx-0.2.21.dist-info/RECORD,,
+bayinx-0.2.22.dist-info/METADATA,sha256=TwM3DxTXPTttN0TGJOPpM1pRMzDvoUjFZWkWWCR9vNI,3058
+bayinx-0.2.22.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+bayinx-0.2.22.dist-info/RECORD,,