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

j0(x)

Implementation of J0 for all x in Jax

j1(x)

Bessel function of order one - using the implementation from CEPHES, translated to Jax.

jaxdem.analysis.bessel.j0(x: Array | ndarray | bool | number | bool | int | float | complex | TypedNdArray) Array[source][source]#

Implementation of J0 for all x in Jax

jaxdem.analysis.bessel.j1(x: Array | ndarray | bool | number | bool | int | float | complex | TypedNdArray) Array[source][source]#

Bessel function of order one - using the implementation from CEPHES, translated to Jax.