Oossa

Amazon向AWS Trainium开放SGLang-Neuron

Amazon现已通过公共容器提供SGLang-Neuron,用户可沿用熟悉的SGLang工作流,在Trainium芯片上运行大语言模型。

简讯作者: Oossa 发布日期: 1 分钟阅读

Amazon宣布,SGLang-Neuron现已通过AWS Deep Learning Container公开提供。该插件将SGLang服务系统与AWS Trainium AI芯片连接起来,用户可以通过熟悉的SGLang API启动模型并生成RL rollout。它基于原生PyTorch运行,并支持混合专家等功能,以及通过Elastic Fabric Adapter传输KV缓存。快速入门指南介绍了如何在Trainium3 EC2实例上启动容器,并提供一个1200亿参数模型的服务。

为什么重要

开发者现在无需更改SGLang代码,就能在AWS的Trainium芯片上运行超大型语言模型,从而降低成本并简化部署。

这篇文章有帮助吗?
分享

继续阅读

Oossa · 新闻简报

一周 AI,讲明白

每周一,用通俗的语言讲清值得知道的新闻。免费,无垃圾邮件。