atomcloud.functions.jax_funcs package

Submodules

• atomcloud.functions.jax_funcs.spence module
  • create_polylog2d()
  • create_spence()

Module contents

Created on Mon Mar 14 19:26:59 2022

@author: hofer

atomcloud.functions.jax_funcs.create_polylog2d(npt)

atomcloud.functions.jax_funcs.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