model_484878007_mixer_nano.py
Model Overview
A nano-scale implementation of the mixer architecture, built for multitask tasks.
Architecture
- Architecture: mixer
- Scale: nano
- Attention: flash
- Fusion strategy: cross attention
- Task head: multitask
- Activation: gelu tanh
- Normalization: rmsnorm
- Initialization: kaiming normal
Training
- Optimizer: novograd
- LR scheduler: constant warmup
Files
model_484878007_mixer_nano.py — main artifact of this repository
License
See the license field above.
