AI Infra 工程师-训练框架优化

让 AI 辅助填写这个岗位的官网网申

使用同一份求职档案填写教育、实习与项目经历；登录、验证码和缺失信息由你处理，最终提交前请本人核对确认。

当前公开岗位库收录亦赋量化 10 个在招校招岗位。岗位数量与状态以企业官网为准。

是否提交成功以企业官网回执或申请记录为准。

工作内容：

• 运用各类手段优化模型训练效率

• 维护和优化高性能 Data Loader

岗位要求：

• 扎实的 Python 编程能力

• 熟悉 Torch

• 熟悉 GPU 架构，了解 Kernel、Stream、Event、Context 等概念

• 掌握使用各类 profiler 工具分析模型的瓶颈点

• 熟悉 GPU 多卡协同通信原理和优化手段

• 熟悉静态图编译技术

• 能运用各类 IO 和并发技术，实现高效的 Data Loader，并持续维护和扩展其功能

加分项：

• 熟悉 Attention 等各类常见模型结构，有能力与算法研究员合作研发更优的模型结构

• 熟悉混合精度训练、模型量化技术，掌握效果调平方法

先核对岗位要求与自己的经历，再准备档案、投递和面试。