← all repositories

lucidrains/mlp-mixer-pytorch

PyTorch implementation of MLP-Mixer, an all-MLP architecture for visual recognition tasks.

1.1k stars Python Computer VisionML Frameworks
mlp-mixer-pytorch
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.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.