强化学习训练框架工程师
岗位摘要
负责强化学习训练框架的研发、优化和维护,根据业务需求持续改进训练框架和策略,提升模型训练效率;分析和定位训练中的性能瓶颈,实施针对性优化措施,提升训练效率和稳定性;跟进业界技术进展,不断同步与集成最新训练优化策略
岗位职责
- 负责强化学习训练框架的研发、优化和维护,根据业务需求持续改进训练框架和策略,提升模型训练效率
- 分析和定位训练中的性能瓶颈,实施针对性优化措施,提升训练效率和稳定性
- 跟进业界技术进展,不断同步与集成最新训练优化策略
任职要求
- 本科及以上学历,计算机相关专业,2-5年工作经验
- 对自然语言处理、计算机视觉和多模态算法有深入理解,熟悉主流的LLM模型架构,有分布式训练经验
- 对常见RL训练算法有基本了解
- 熟悉vllm或sglang等常用开源推理框架的优先考虑