Obecnie obsługujemy te jądra GPU:
tokamax.dot_product_attention
(FlashAttention).tokamax.gated_linear_unit(liniowe jednostki z bramką, np. SwiGLU).tokamax.layer_norm(normalizacja warstwowa i normalizacja średniej kwadratowej).W przypadku GPU i TPU:
tokamax.ragged_dot
(Mixture of Experts).O ile nie stwierdzono inaczej, treść tej strony jest objęta licencją Creative Commons – uznanie autorstwa 4.0, a fragmenty kodu są dostępne na licencji Apache 2.0. Szczegółowe informacje na ten temat zawierają zasady dotyczące witryny Google Developers. Java jest zastrzeżonym znakiem towarowym firmy Oracle i jej podmiotów stowarzyszonych.
Ostatnia aktualizacja: 2025-11-21 UTC.