As you increase like the number of nodes by a factor of two say you're roughly increasing the number of edges in the graph by a Factor of four because these networks are highly connected. Your layer zero heads have an impact on almost all downstream layer heads so that turns out to be the biggest bottleneck. Because this algorithm is iterative over each of the edges this leads to like yeah like the quadratic increase and then like your whole forward pass is now more expensive as well but i don't think would give you any insight into how it works.

Get the Snipd
podcast app

Unlock the knowledge in podcasts with the podcast player of the future.
App store bannerPlay store banner

AI-powered
podcast player

Listen to all your favourite podcasts with AI-powered features

Discover
highlights

Listen to the best highlights from the podcasts you love and dive into the full episode

Save any
moment

Hear something you like? Tap your headphones to save it with AI-generated key takeaways

Share
& Export

Send highlights to Twitter, WhatsApp or export them to Notion, Readwise & more

AI-powered
podcast player

Listen to all your favourite podcasts with AI-powered features

Discover
highlights

Listen to the best highlights from the podcasts you love and dive into the full episode