负责蚂蚁强化学习引擎开发
职位描述
1. 精通至少一门 Python/Go/C++ 等编程语言,并有良好的代码风格;
2. 具有分布式系统服务/并行计算系统设计与研发优化经验;
3. 代码级精通主流深度学习框架及扩展库的使用及算子开发,例如TensorFlow/PyTorch/Megatron/Deepspeed/vLLM/Sglang等;
4. 好奇心强,热爱技术且对强化学习领域有深入钻研优先;
加分项:
1、熟悉至少一种主流的RLHF框架,如OpenRLHF/veRL/AReal/ChatLearn等;
2、熟悉Ray框架或其他强化学习相关计算框架;
3、在计算机系统网络顶会OSDI/SOSP/NSDI/ATC/EuroSys上有文章发表经验