Fix: Missing Python APIs for probability dists

In one of the previous commits the `kullbackleibler`
and `jensenshannon` interfaces were removed.

Author: ashvardanian

python/bench.py

@@ -212,8 +212,8 @@ def raise_(ex):
                 [np.float64, np.float32, np.float16],
                 np.array,
             ),
-            ("scipy.hamming", spd.hamming, simd.hamming, [np.uint8], np.array),
-            ("scipy.jaccard", spd.jaccard, simd.jaccard, [np.uint8], np.array),
+            ("scipy.hamming", spd.hamming, lambda a, b: simd.hamming(a, b, "b8"), [np.uint8], np.array),
+            ("scipy.jaccard", spd.jaccard, lambda a, b: simd.jaccard(a, b, "b8"), [np.uint8], np.array),
         ]
     )
 
@@ -348,8 +348,8 @@ def raise_(ex):
                 simd.kullbackleibler,
                 [np.float64, np.float32, np.float16],
             ),
-            ("scipy.hamming", wrap_rowwise(spd.hamming), simd.hamming, [np.uint8]),
-            ("scipy.jaccard", wrap_rowwise(spd.jaccard), simd.jaccard, [np.uint8]),
+            ("scipy.hamming", wrap_rowwise(spd.hamming), lambda a, b: simd.hamming(a, b, "b8"), [np.uint8]),
+            ("scipy.jaccard", wrap_rowwise(spd.jaccard), lambda a, b: simd.jaccard(a, b, "b8"), [np.uint8]),
         ]
     )
 

python/lib.c

@@ -1236,6 +1236,36 @@ static PyMethodDef simsimd_methods[] = {
         "Notes:\n"
         "    * `a` and `b` are positional-only arguments, while `dtype` is a keyword-only argument.",
     },
+    {
+        "kullbackleibler",
+        (PyCFunction)api_kl,
+        METH_FASTCALL,
+        "Compute Kullback-Leibler divergences between two matrices.\n\n"
+        "Args:\n"
+        "    a (NDArray): First floating-point matrix or vector.\n"
+        "    b (NDArray): Second floating-point matrix or vector.\n"
+        "    dtype (IntegralType, optional): Override the presumed input type.\n\n"
+        "Returns:\n"
+        "    DistancesTensor: The Kullback-Leibler divergences distances.\n\n"
+        "Equivalent to: `scipy.special.kl_div`.\n"
+        "Notes:\n"
+        "    * `a` and `b` are positional-only arguments, while `dtype` is a keyword-only argument.",
+    },
+    {
+        "jensenshannon",
+        (PyCFunction)api_js,
+        METH_FASTCALL,
+        "Compute Jensen-Shannon divergences between two matrices.\n\n"
+        "Args:\n"
+        "    a (NDArray): First floating-point matrix or vector.\n"
+        "    b (NDArray): Second floating-point matrix or vector.\n"
+        "    dtype (IntegralType, optional): Override the presumed input type.\n\n"
+        "Returns:\n"
+        "    DistancesTensor: The Jensen-Shannon divergences distances.\n\n"
+        "Equivalent to: `scipy.spatial.distance.jensenshannon`.\n"
+        "Notes:\n"
+        "    * `a` and `b` are positional-only arguments, while `dtype` is a keyword-only argument.",
+    },
 
     // Conventional `cdist` interface for pairwise distances
     {