专栏名称: 爱可可-爱生活
知名互联网资讯博主 北邮PRIS模式识别实验室陈老师
目录
今天看啥  ›  专栏  ›  爱可可-爱生活

【JetStream:专为XLA设备优化的LLM推理引擎,针对T-20241205212054

爱可可-爱生活  · 微博  · AI  · 2024-12-05 21:20
    

文章预览

2024-12-05 21:20 本条微博链接 【JetStream:专为XLA设备优化的LLM推理引擎,针对TPU和GPU进行了吞吐量和内存优化,支持JAX和PyTorch模型,提供完整的服务部署方案。特别适合在Cloud TPU VM上进行在线推理,可用于Gemma等大模型的高效部署】 'JetStream is a throughput and memory optimized engine for LLM inference on XLA devices, starting with TPUs (and GPUs in future -- PRs welcome).' GitHub: github.com/AI-Hypercomputer/JetStream #大模型推理# #TPU优化# #MLOps# #LLM部署# ………………………………

原文地址:访问原文地址
快照地址: 访问文章快照
总结与预览地址:访问总结与预览