От тех же ребят из стэнфорда мне понравилась идея мега-кернелов (такое как-то было в комп графике) https://hazyresearch.stanford.edu/blog/2025-05-27-no-bubbles мы всю или почти всю модель запекаем в один огромный кернел и реплееим его без цпу со скоростью света