微信支付-大模型 AI Infra 研发工程师

岗位职责

1. 负责微信支付大模型推理平台建设与性能优化，结合模型结构、业务负载和硬件特性开展软硬件协同优化，持续提升推理吞吐、降低时延，保障大规模异构GPU集群稳定运行；
2. 负责大模型训练性能优化，面向SFT、强化学习等任务，优化显存管理、分布式并行、通信和资源调度，提升训练效率及算力利用率；
3. 负责国产GPU适配与异构算力优化，包括算子适配、图优化、编译优化及集合通信调优，推动主流模型和训推框架稳定落地。

任职要求

1. 熟悉vLLM、SGLang等推理框架，理解请求调度、KV Cache、Prefill-Decode分离、长上下文推理、Continuous Batching及投机解码等技术；
2. 熟悉DeepSpeed、Megatron-LM、verl、LLaMA-Factory等训练及分布式框架，理解数据并行、张量并行、流水线并行和专家并行，以及显存、计算与通信优化；
3. 熟悉GPU架构及软件栈，具备算子、图编译或集合通信优化经验；有国产GPU适配、CUDA开发、高性能算子开发、AI编译器或NCCL优化经验者优先；
4. 熟悉DeepSeek、混元、Qwen、GLM、Kimi等主流模型的结构与部署，在训练或推理至少一个方向有深入实践；训推兼备，或具有大规模GPU集群训推优化及资源管理经验者优先；
5. 具备良好的沟通协作、问题分析和快速学习能力，自我驱动，对前沿技术有热情；有知名开源项目开发维护经历，或在ACM等知名编程竞赛中获奖者优先。

产品/项目

微信支付

工作年限

三年以上工作经验