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
- Code: GitHub - sayakpaul/MLPMixer-jax2tf: This repository hosts code for converting the original MLP Mixer models (JAX) to TensorFlow.
Here is some example usage:
References:
[1] MLP-Mixer: An all-MLP Architecture for Vision by Tolstikhin et al.
