OpenXLA Project

Questa pagina è stata tradotta dall'API Cloud Translation.

Operazioni e hardware supportati

Al momento supportiamo i seguenti kernel GPU:

tokamax.dot_product_attention (FlashAttention).
tokamax.gated_linear_unit (Unità lineari controllate (SwiGLU e così via)).
tokamax.layer_norm (normalizzazione dei livelli e normalizzazione della radice quadrata media).

E quanto segue sia per la GPU che per la TPU:

Salvo quando diversamente specificato, i contenuti di questa pagina sono concessi in base alla licenza Creative Commons Attribution 4.0, mentre gli esempi di codice sono concessi in base alla licenza Apache 2.0. Per ulteriori dettagli, consulta le norme del sito di Google Developers. Java è un marchio registrato di Oracle e/o delle sue consociate.

Ultimo aggiornamento 2025-11-21 UTC.