lucidrains/mlp-mixer-pytorch
PyTorch implementation of MLP-Mixer, an all-MLP architecture for visual recognition tasks.

Velocity · 7d
+0.6
★ / day
Trend
→steady
star history
This repository provides a PyTorch implementation of the MLP-Mixer architecture, an all-MLP solution for vision introduced by Google AI. The model processes images and videos as patches, using only multi-layer perceptrons without convolutions or attention mechanisms. It supports both 2D image classification and 3D video classification through separate model variants.