Oossa

i14 addestra Laplace 3B su AWS Trainium2 a 120.000 token/s

Il modello ad attenzione lineare da 3 miliardi di parametri è stato addestrato a 120.000 token al secondo e ha generato 11.544 token al secondo su un’istanza trn2.48xlarge.

BreveDi Pubblicato da Oossa: 1 min di lettura

i14 ha usato AWS Trainium2 e TorchNeuron Native per addestrare e servire il suo modello Laplace 3B ad attenzione lineare. Su una singola istanza trn2.48xlarge, il modello è stato addestrato a circa 120.000 token al secondo e ha generato 11.544 token al secondo su 1.280 sequenze simultanee. L’architettura mantiene lo stato di inferenza fisso a 29,5 MB per sequenza, indipendentemente dalla lunghezza del contesto.

Perché conta

Un uso prevedibile della memoria consente agli agenti IA di lunga durata di elaborare testi molto più ampi senza far aumentare i costi.

Questo articolo ti è stato utile?
Condividi

Leggi anche

Oossa · Newsletter

La settimana dell'IA, spiegata

Ogni lunedì: le notizie che contano, in parole semplici. Gratis, niente spam.