pytensor 2.26.2__tar.gz → 2.26.3__tar.gz

{pytensor-2.26.2/pytensor.egg-info → pytensor-2.26.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pytensor
-Version: 2.26.2
+Version: 2.26.3
 Summary: Optimizing compiler for evaluating mathematical expressions on CPUs and GPUs.
 Author-email: pymc-devs <pymc.devs@gmail.com>
 License: .. _license:

{pytensor-2.26.2 → pytensor-2.26.3}/pytensor/_version.py

@@ -8,11 +8,11 @@ import json
 
 version_json = '''
 {
- "date": "2024-11-11T14:51:54+0100",
+ "date": "2024-11-13T15:48:28+0100",
  "dirty": false,
  "error": null,
- "full-revisionid": "fdbf3aa56a2f2e0e1bb8dc97816c8d0ba34ca5ed",
- "version": "2.26.2"
+ "full-revisionid": "33a4d4882cc14c1a4a24ec24a5ebb7a1065bcbae",
+ "version": "2.26.3"
 }
 '''  # END VERSION_JSON
 

{pytensor-2.26.2 → pytensor-2.26.3}/pytensor/graph/basic.py

@@ -616,16 +616,20 @@ class Variable(Node, Generic[_TypeType, OptionalApplyType]):
         """
         from pytensor.compile.function import function
 
+        ignore_unused_input = kwargs.get("on_unused_input", None) in ("ignore", "warn")
+
         def convert_string_keys_to_variables(inputs_to_values) -> dict["Variable", Any]:
             new_input_to_values = {}
             for key, value in inputs_to_values.items():
                 if isinstance(key, str):
                     matching_vars = get_var_by_name([self], key)
                     if not matching_vars:
-                        raise ValueError(f"{key} not found in graph")
+                        if not ignore_unused_input:
+                            raise ValueError(f"{key} not found in graph")
                     elif len(matching_vars) > 1:
                         raise ValueError(f"Found multiple variables with name {key}")
-                    new_input_to_values[matching_vars[0]] = value
+                    else:
+                        new_input_to_values[matching_vars[0]] = value
                 else:
                     new_input_to_values[key] = value
             return new_input_to_values

{pytensor-2.26.2 → pytensor-2.26.3}/pytensor/link/c/cmodule.py

@@ -2379,6 +2379,14 @@ class GCC_compiler(Compiler):
         if sys.platform == "darwin":
             # Use the already-loaded python symbols.
             cxxflags.extend(["-undefined", "dynamic_lookup"])
+            # XCode15 introduced ld_prime linker. At the time of writing, this linker
+            # leads to multiple issues, so we supply a flag to use the older dynamic
+            # linker: ld64
+            if int(platform.mac_ver()[0].split(".")[0]) >= 15:
+                # This might be incorrect. We know that ld_prime was introduced in
+                # XCode15, but we don't know if the platform version is aligned with
+                # xcode's version.
+                cxxflags.append("-ld64")
 
         if sys.platform == "win32":
             # Workaround for https://github.com/Theano/Theano/issues/4926.

{pytensor-2.26.2 → pytensor-2.26.3}/pytensor/link/jax/dispatch/scalar.py

@@ -31,6 +31,7 @@ from pytensor.scalar.math import (
     GammaIncInv,
     Iv,
     Ive,
+    Kve,
     Log1mexp,
     Psi,
     TriGamma,
@@ -288,9 +289,12 @@ def jax_funcify_Iv(op, **kwargs):
 
 @jax_funcify.register(Ive)
 def jax_funcify_Ive(op, **kwargs):
-    ive = try_import_tfp_jax_op(op, jax_op_name="bessel_ive")
+    return try_import_tfp_jax_op(op, jax_op_name="bessel_ive")
+
 
-    return ive
+@jax_funcify.register(Kve)
+def jax_funcify_Kve(op, **kwargs):
+    return try_import_tfp_jax_op(op, jax_op_name="bessel_kve")
 
 
 @jax_funcify.register(Log1mexp)

{pytensor-2.26.2 → pytensor-2.26.3}/pytensor/scalar/math.py

@@ -1281,6 +1281,38 @@ class Ive(BinaryScalarOp):
 ive = Ive(upgrade_to_float, name="ive")
 
 
+class Kve(BinaryScalarOp):
+    """Exponentially scaled modified Bessel function of the second kind of real order v."""
+
+    nfunc_spec = ("scipy.special.kve", 2, 1)
+
+    @staticmethod
+    def st_impl(v, x):
+        return scipy.special.kve(v, x)
+
+    def impl(self, v, x):
+        return self.st_impl(v, x)
+
+    def L_op(self, inputs, outputs, output_grads):
+        v, x = inputs
+        [kve_vx] = outputs
+        [g_out] = output_grads
+        # (1 -v/x) * kve(v, x) - kve(v - 1, x)
+        kve_vm1x = self(v - 1, x)
+        dx = (1 - v / x) * kve_vx - kve_vm1x
+
+        return [
+            grad_not_implemented(self, 0, v),
+            g_out * dx,
+        ]
+
+    def c_code(self, *args, **kwargs):
+        raise NotImplementedError()
+
+
+kve = Kve(upgrade_to_float, name="kve")
+
+
 class Sigmoid(UnaryScalarOp):
     """
     Logistic sigmoid function (1 / (1 + exp(-x)), also known as expit or inverse logit

{pytensor-2.26.2 → pytensor-2.26.3}/pytensor/tensor/math.py

@@ -1229,6 +1229,16 @@ def ive(v, x):
     """Exponentially scaled modified Bessel function of the first kind of order v (real)."""
 
 
+@scalar_elemwise
+def kve(v, x):
+    """Exponentially scaled modified Bessel function of the second kind of real order v."""
+
+
+def kv(v, x):
+    """Modified Bessel function of the second kind of real order v."""
+    return kve(v, x) * exp(-x)
+
+
 @scalar_elemwise
 def sigmoid(x):
     """Logistic sigmoid function (1 / (1 + exp(-x)), also known as expit or inverse logit"""
@@ -3040,6 +3050,8 @@ __all__ = [
     "i1",
     "iv",
     "ive",
+    "kv",
+    "kve",
     "sigmoid",
     "expit",
     "softplus",

{pytensor-2.26.2 → pytensor-2.26.3}/pytensor/tensor/rewriting/math.py

@@ -56,6 +56,7 @@ from pytensor.tensor.math import (
     ge,
     int_div,
     isinf,
+    kve,
     le,
     log,
     log1mexp,
@@ -3494,3 +3495,18 @@ local_polygamma_to_tri_gamma = PatternNodeRewriter(
 )
 
 register_specialize(local_polygamma_to_tri_gamma)
+
+
+local_log_kv = PatternNodeRewriter(
+    # Rewrite log(kv(v, x)) = log(kve(v, x) * exp(-x)) -> log(kve(v, x)) - x
+    # During stabilize -x is converted to -1.0 * x
+    (log, (mul, (kve, "v", "x"), (exp, (mul, -1.0, "x")))),
+    (sub, (log, (kve, "v", "x")), "x"),
+    allow_multiple_clients=True,
+    name="local_log_kv",
+    # Start the rewrite from the less likely kve node
+    tracks=[kve],
+    get_nodes=get_clients_at_depth2,
+)
+
+register_stabilize(local_log_kv)

{pytensor-2.26.2 → pytensor-2.26.3/pytensor.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: pytensor
-Version: 2.26.2
+Version: 2.26.3
 Summary: Optimizing compiler for evaluating mathematical expressions on CPUs and GPUs.
 Author-email: pymc-devs <pymc.devs@gmail.com>
 License: .. _license: