职位描述
- Python
- C/C++
1. 参与公司训练底层框架研发, 负责超大规模训练技术在公司实际业务场景中的计算效率优化.
2. 超大规模训练场景下,探索前沿的分布式系统设计,包括不限于通信优化、内存优化、计算优化等。
3. 参与建设 PyTorch 2.0/TorchDynamo/TorchInductor/Triton 等编译技术优化.
职位要求
1. 扎实系统功底和编程能力,熟悉 C++/Python,熟悉常见数据结构和设计模式;
2. 熟悉深度学习基本原理和 AI 框架 (TensorFlow/PyTorch/JAX等);
3. 拥有高性能计算或模型优化经验和兴趣 (CUDA/Triton/OpenMP);
4. 有 AI 编译器框架或社区相关经验 (如MLIR、TVM、XLA等);
5. 热爱技术、有较强的学习能力、复杂问题归纳梳理能力.
认证资质
王女士
竞争力分析
BOSS 安全提示
BOSS直聘严禁用人单位和招聘者用户做出任何损害求职者合法权益的违法违规行为,包括但不限于扣押求职者证件、收取求职者财物、向求职者集资、让求职者入股、诱导求职者异地入职、异地参加培训、违法违规使用求职者简历等,您一旦发现此类行为, 请立即举报
了解更多职场安全防范知识精选职位
更新于:2024-11-01