@@ -321,8 +321,23 @@ def k1_large(x):
321321
322322# =====================================================================
323323# Modified Bessel K_v(x) for fractional order
324- # Ported from TensorFlow Probability's bessel.py (Apache 2.0 License)
325- # Uses Temme's method (|v| < 50) + Olver's uniform asymptotic (|v| >= 50)
324+ #
325+ # The functions below (_sqrt1px2, _evaluate_temme_coeffs,
326+ # _temme_series_kve, _continued_fraction_kve, _olver_kve, _temme_kve,
327+ # _kve_core) and the _ASYMPTOTIC_OLVER_COEFFICIENTS constant are
328+ # derived from TensorFlow Probability's bessel.py:
329+ # https://github.com/tensorflow/probability
330+ #
331+ # Original copyright and license:
332+ # Copyright 2020 The TensorFlow Probability Authors.
333+ # Licensed under the Apache License, Version 2.0.
334+ #
335+ # Modifications from the original:
336+ # - Ported from TensorFlow/TFP APIs to pure JAX (jax.numpy, jax.lax)
337+ # - Removed I_v (bessel_ive) computation; only K_v is computed
338+ # - Removed log-space output option
339+ # - Removed negative-v correction for I_v
340+ # - Simplified to scalar-only core (vectorization via jax.vmap)
326341# =====================================================================
327342
328343# Olver expansion polynomial coefficients (10 terms, up to 31 coefficients each)
0 commit comments