rishikksh20 / FNet-pytorch

Unofficial implementation of Google's FNet: Mixing Tokens with Fourier Transforms
MIT License
251 stars 37 forks source link
feedforward-neural-network fnet fourier-transform image-classification language-model text text-classification transformer vision

FNet: Mixing Tokens with Fourier Transforms

Pytorch implementation of Fnet : Mixing Tokens with Fourier Transforms.

Citation:

@misc{leethorp2021fnet,
      title={FNet: Mixing Tokens with Fourier Transforms}, 
      author={James Lee-Thorp and Joshua Ainslie and Ilya Eckstein and Santiago Ontanon},
      year={2021},
      eprint={2105.03824},
      archivePrefix={arXiv},
      primaryClass={cs.CL}
}