Discussion about this post

User's avatar
Yaroslav Bulatov's avatar

Here's a fun experiment with architecture co-design -- https://github.com/thomasnormal/spicenn2 . Here's another one for GPUs https://github.com/cybertronai/sutro-problems/tree/main/mnist

Yaroslav Bulatov's avatar

There's also architecture co-design. Hardware people usually optimize pre-trained models to their hardware, but these pretrained models were designed for GPUs so that limits your upside. Retrained with new arch is painful, but maybe distillation from OSS checkpoint makes it feasible. I feel like Taalas got a lucky with the demo, they have something like 6k context, realistic context would be >10GB KV-cache, which sucks for chips without HBM (https://chatgpt.com/share/6aa5cd64-42b4-83e8-8e89-90c4cc36d52a)

No posts

Ready for more?