AI Infra工程师

AI Infra工程师上海正式软件类职位描述1. 负责团队内 3D 视觉、视频生成、神经渲染等方向模型的训练与推理工程化落地，将算法原型转化为稳定、高效、可维护的生产级系统 2. 参与大规模分布式训练与推理平台建设，优化数据处理、模型管理与评测流程，提升研发迭代效率 3. 基于 C++/CUDA 进行核心算子与热点路径性能优化，针对大规模多模态场景做显存与吞吐优化 4. 参与推理服务搭建与部署，包括模型导出、加速优化、批处理与并发调度，保障服务稳定性与可观测性 5. 开发内部工程工具链，包括实验管理、性能评测、自动化测试与 CI/CD 集成，推动工程质量持续提升 6. 与算法团队紧密协作，响应训练集群与线上服务的性能与稳定性问题职位要求1. 计算机、软件工程、电子信息、自动化等相关专业本科及以上学历，2025年9月至2026年8月毕业（或已毕业） 2. 工程实现能力强，熟练掌握 C/C++、Python，代码风格良好，有完整项目或实习工程经验 3. 熟悉 PyTorch 训练与推理流程，了解 Linux 开发环境、Git、Docker，有 GPU 集群使用经验 4. 满足以下方向之一：有 CUDA/GPU 性能优化经验；有分布式训练或多机部署经验；有推理服务或高并发系统开发经验 5. 对 AI 基础设施与工程化落地有强烈兴趣，沟通顺畅，结果导向，能与算法同学高效协作 6. 加分项：有算子开发或深度学习框架扩展经验 7. 加分项：参与过多模态大模型、3D 视觉或视频生成相关工程项目 8. 加分项：熟悉主流训练推理加速技术栈，或有 CI/CD、监控告警实践经验 9. 加分项：编程竞赛获奖，或参与过知名开源项目贡献投递