I want that pixel! An implementation of Slot Attention in JAX
6 january 2026 · 6 min read
The mechanism of Attention, widely popularized by the advent of ChatGPT and subsequent Large Language Models is one of the most beautiful and elegant mechanisms that have revolutionized the world. …