Add numba implementation for gammaln

aesara/link/numba/dispatch/scalar.py

@@ -27,7 +27,7 @@ from aesara.scalar.basic import (
     Second,
     Switch,
 )
-from aesara.scalar.math import Sigmoid
+from aesara.scalar.math import Sigmoid, GammaLn
 
 
 @numba_funcify.register(ScalarOp)
@@ -251,3 +251,14 @@ def numba_funcify_Sigmoid(op, node, **kwargs):
         return 1 / (1 + np.exp(-x))
 
     return sigmoid
+
+
+@numba_funcify.register(GammaLn)
+def numba_funcify_Sigmoid(op, node, **kwargs):
+    import math
+
+    @numba_basic.numba_njit(inline="always", fastmath=config.numba__fastmath)
+    def gammaln(x):
+        return math.lgamma(x)
+
+    return gammaln

tests/link/test_numba.py

@@ -323,8 +323,8 @@ def test_box_unbox(input, wrapper_fn, check_fn):
     "inputs, input_vals, output_fn, exc",
     [
         (
-            [at.lvector()],
-            [rng.poisson(10, size=100).astype(np.int64)],
+            [at.vector()],
+            [rng.uniform(size=100).astype(config.floatX)],
             lambda x: at.gammaln(x),
             None,
         ),