fnet-flax (pypi): Minimal Flax implementation of FNet: Mixing Tokens with Fourier Transforms
Keywords: deep-learning|flax|jax|machine-learning|neural-networks|pypi|pypi-package|python