AI Infra 训练优化工程师

工作内容：

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

• 维护和优化高性能 Data Loader

岗位要求：

• 扎实的 Python 编程能力

• 熟悉 Torch

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

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

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

• 熟悉静态图编译技术

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

加分项：

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

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