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

岗位描述

1. 负责AI4S模型（如MLFF、Cofolding等）的训练与推理性能优化，提升模型吞吐、资源利用率和整体计算效率；
2. 基于PyTorch、CUDA、Triton等技术栈，分析和解决模型训练、推理过程中的性能瓶颈；
3. 利用Torch.Compile、编译优化及算子融合等技术，持续提升模型执行效率和GPU利用率；
4. 建立性能分析与Benchmark体系，使用Profiling工具定位系统热点并推动优化方案…
5. 硕士学位及以上，计算机、软件工程、数学、物理等相关专业，3年以上AI系统性能优化工作经验；
6. 熟练掌握Python和C++，具备扎实的软件工程能力和良好的编码习惯；
7. 熟悉…