蚂蚁集团
网商银行-AI-Infra工程师-推训方向
RESPONSIBILITIES
岗位职责
负责大模型分布式训练与推理框架的研发和性能优化,持续提升训练效率、推理吞吐、资源利用率及系统稳定性;
与算法团队紧密协作,针对新模型结构(Dense/MoE等)提供框架层面的支持与适配,支撑算法高效迭代与算力高效利用;
建设推理服务的可观测性与容错能力,完善监控、诊断、限流、降级与故障恢复机制,保障线上服务 SLA;
调研、评估并推动 SGLang、vLLM、TensorRT-LLM、FlashInfer、Triton Inference Server 等前沿推理框架及优化技术在内部落地;
参与GPU/异构集群的资源调度与成本优化,提升整体算力利用效率;
REQUIREMENTS
任职要求
5年及以上 AI Infra、AI 系统、高性能计算或分布式系统相关研发经验;
熟悉GPU 体系结构与 CUDA 编程,了解国产加速芯片(昇腾 NPU、平头哥 PPU 等)及配套软件栈者优先;
熟悉SGLang、vLLM、TensorRT-LLM 等至少一种主流推理框架,有实际性能优化经验;
具备分布式通信(NCCL、RDMA)及 Linux 系统级性能分析能力;
良好的工程素养与协作能力,能将技术方案落地为稳定、可维护的工程产物;
信息来自企业官方招聘渠道
OfferSeek 对公开岗位信息进行聚合、去重和结构化整理,最终申请以企业官方页面为准。