jaxdem.analysis.bessel#
Bessel functions for JAX.
Originally adapted from: https://github.com/benjaminpope/sibylla/blob/main/notebooks/bessel_test.ipynb
Note: This module does not change global JAX configuration (e.g. x64 enablement). If you want 64-bit execution, set it in your application before importing JaxDEM:
>>> import jax
>>> jax.config.update("jax_enable_x64", True)
Functions
|
Implementation of J0 for all x in Jax |
|
Bessel function of order one - using the implementation from CEPHES, translated to Jax. |