mlpmixer-flax (pypi): Flax implementation of MLP-Mixer from MLP-Mixer: An all-MLP Architecture for Vision
Keywords: deep-learning|flax|jax|machine-learning|neural-networks|pypi|pypi-package|python