Oossa

i14 traint Laplace 3B op AWS Trainium2 met 120.000 tokens per seconde

Het model met 3 miljard parameters en lineaire aandacht werd getraind met 120.000 tokens per seconde en verwerkte 11.544 tokens per seconde op een trn2.48xlarge-instantie.

Kort berichtDoor Gepubliceerd door Oossa: 1 min lezen

i14 gebruikte AWS Trainium2 en TorchNeuron Native om het Laplace 3B-model met lineaire aandacht te trainen en uit te voeren. Op één trn2.48xlarge-instantie werd het model getraind met ongeveer 120.000 tokens per seconde en kon het 11.544 outputtokens per seconde genereren voor 1.280 gelijktijdige reeksen. De architectuur houdt de inferentiestatus per reeks vast op 29,5 MB, ongeacht de lengte van de context.

Waarom het ertoe doet

Dankzij het voorspelbare geheugengebruik kunnen AI-agents die lang actief blijven veel grotere teksten verwerken zonder dat de kosten oplopen.

Was dit artikel nuttig?
Delen

Lees ook

Oossa · Nieuwsbrief

De AI-week, uitgelegd

Elke maandag: het nieuws dat ertoe doet, in gewone taal. Gratis, geen spam.