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芯片上运行超大型语言模型,从而降低成本并简化部署。