i14 trains and serves Laplace 3B linear-attention model on AWS Trainium2 with TorchNeuron Native
i14 ported its Laplace 3B 3B-parameter linear-attention decoder (using Gated Linear Attention and a fixed 29.5 MB recurrent state) to AWS Trainium2 via TorchNeuron Native and native PyTorch operators. On a trn2.48xlarge instance the team reported ~120,000 training tokens/sec and serving throughput of 11,544 output tokens/sec across 1,280 concurrent sequences after replacing a Triton kernel, optimizing MuonFSDP2 collectives, and statically unrolling the GLA recurrence for the Trainium2 compiler.