AI for Science性能优化工程师-Seed

【岗位职责】 1、负责AI4S模型（如MLFF、Cofolding等）的训练与推理性能优化，提升模型吞吐、资源利用率和整体计算效率； 2、基于PyTorch、CUDA、Triton等技术栈，分析和解决模型训练、推理过程中的性能瓶颈； 3、利用Torch.Compile、编译优化及算子融合等技术，持续提升模型执行效率和GPU利用率； 4、建立性能分析与Benchmark体系，使用Profiling工具定位系统热点并推动优化方案落地； 5、与算法、平台及科学计算团队紧密协作，共同推动AI4S模型和基础设施的性能演进。 【任职要求】 1、硕士学位及以上，计算机、软件工程、数学、物理等相关专业，3年以上AI系统性能优化工作经验； 2、熟练掌握Python和C++，具备扎实的软件工程能力和良好的编码习惯； 3、熟悉PyTorch训练与推理流程，具备模型性能优化的实际项目经验； 4、熟悉CUDA编程…… 完整岗位描述与官方投递入口,请登录后在岗位详情页查看。

本站当前在招 19019 个北京岗位（全城各职能合计），其中 2814 个公开标注了薪资。按这些标注统计：月薪（税前）主流区间 7K ~ 20K（25~75 分位），中位约 9K，最低 2K、最高 300K。