As of today, different variants of MLP-Mixers [1] are now available on TensorFlow Hub. Below are the details:


Models: https://tfhub.dev/sayakpaul/collections/mlp-mixer/1 10


Code: https://git.io/JzR68 7


Here is some example usage:

References:
[1] MLP-Mixer: An all-MLP Architecture for Vision by Tolstikhin et al. 4