Official implementaiton of RefAM: Attention Magnets for Zero-Shot Referral Segmentaiton
-
Updated
Feb 6, 2026 - Jupyter Notebook
Official implementaiton of RefAM: Attention Magnets for Zero-Shot Referral Segmentaiton
Faithful from-scratch PyTorch reproduction of OpenAI's GPT-OSS architecture (sliding/full attention alternation, learned attention sinks, YaRN 128K, top-2-of-8 MoE), scaled to Chinchilla-optimal 502M total / 247M active training on a single A100 80GB
Add a description, image, and links to the attention-sinks topic page so that developers can more easily learn about it.
To associate your repository with the attention-sinks topic, visit your repo's landing page and select "manage topics."