5.52. extras.jax

Implementation of operators using Jax FFT functions.

For more details see operators.

For installation instructions see https://jax.readthedocs.io/en/latest/installation.html.

Alternatively, one can try the default installation with:

pip install eminus[jax]

This implementation is focused on speed, rather than readability since these operators need the most time in most calculations, see torch.

Note that in testings this backend was always slower than the torch or scipy backend.

Reference: https://github.com/google/jax.

Functions

I	Backwards transformation from reciprocal space to real-space.
Idag	Conjugated backwards transformation from real-space to reciprocal space.
J	Forward transformation from real-space to reciprocal space.
Jdag	Conjugated forward transformation from reciprocal space to real-space.