The scipy subroutines for the spherical Bessel functions and Bessel functions (like scipy.special.spherical_jn) are not included in the JAX library. Although other Bessel functions are included within JAX, the spherical Bessel functions are very important in a variety of applications, such as the Fourier Transform of a function in three-dimensions.
Although the scipy library in JAX does have some modified Bessel function subroutines, expanding it to include the entire Bessel function/Spherical Bessel function subroutines in scipy.special would be extremely helpful.
The scipy subroutines for the spherical Bessel functions and Bessel functions (like scipy.special.spherical_jn) are not included in the JAX library. Although other Bessel functions are included within JAX, the spherical Bessel functions are very important in a variety of applications, such as the Fourier Transform of a function in three-dimensions.
Although the scipy library in JAX does have some modified Bessel function subroutines, expanding it to include the entire Bessel function/Spherical Bessel function subroutines in scipy.special would be extremely helpful.