Beyond the KV cache: Training and serving i14’s Laplace 3B linear attention model on AWS Trainium2 with native PyTorch
Most large language models (LLMs) use transformer architectures with softmax attention. During inference, their key-value (KV) cache grows with the…
