I think that we need to better promote our TF models in paperswithcode /cc @Joana.
In this specific case you can see the official Google reference  implementation in JAX and all the others alternative implementations:

  

      paperswithcode.com
  

  
    

Papers with Code - MLP-Mixer: An all-MLP Architecture for Vision 15

  #11 best model for Image Classification on ImageNet ReaL (Accuracy metric)