atomcloud.functions.jax_funcs.spence module

atomcloud.functions.jax_funcs.spence.create_polylog2d(npt)

atomcloud.functions.jax_funcs.spence.create_spence(npt)

Create a function that calculates the spence function using either scipy.special.spence or a jaxified version of the same function. I ported the Cephes library algorithm to jax, but which is equivalent for real numbers.

Parameters:

npt – numpy or jax

Returns:

spence function