i14 推出 Laplace 3B 線性注意力模型,於 AWS Trainium2 上以原生 PyTorch 執行訓練與推論
Beyond the KV cache: Training and serving i14’s Laplace 3B linear attention model on AWS Trainium2 with native PyTorch
AI 導讀
i14 的 Laplace 3B 線性注意力模型已在 AWS Trainium2 上完成訓練與推論,並以原生 PyTorch 取得 120,000 tokens/s 的訓練吞吐與 11,544 tokens/s 的推論吞吐。
信源出處aws.amazon.com
Beyond the KV cache: Training and serving i14’s Laplace 3B linear attention model on AWS Trainium2 with native PyTorch
本文為 888news AI 精選之重點導讀。若需閱讀作者完整原文、技術細節與官方討論,請前往原始網站。
來源: AWS Machine Learning Blog (aws.amazon.com)
開啟原文