keras
keras-fft

[jax]
jax
jaxlib

[tensorflow]
tensorflow
