search

amimre / spbessax

Spherical bessel functions for JAX
Apache License 2.0
3 stars 0 forks source link

Any plans to add iv? #1

Open NeilGirdhar opened 1 month ago

NeilGirdhar commented 1 month ago

Specifically,

scipy.special.iv
scipy.special.ive
tensorflow_probability.math.bessel_iv_ratio
tensorflow_probability.math.log_bessel_ive

You could just lift the code from here.

amimre commented 1 month ago

Seems feasible, I will look into it!

NeilGirdhar commented 1 month ago

Awesome! I did about half the work of porting from Tensorflow to simple Jax here: https://github.com/NeilGirdhar/tjax/tree/bessel

Half the tests pass, the other half need some investigation. I don't plan on continuing, but this might save you some time :smile: