25届春招-后端开发工程师(机器学习算法工程方向)

薪资面议
H88越虹广场B座3楼
2025-02-17










岗位职责
1、负责预训练模型的训练加速。
2、负责预训练模型的部署加速。包括但不限于多卡任务调度,算子优化等;
3、结合业务以及平台化需求,参与大模型数据、训练、调优、部署平台开发;
任职条件
1、本科以上学历,计算机相关专业,三年以上深度学习模型开发经验;
2、熟练使用CUDA/C++/Python等相关工具,对DeepSpeed,PyTorch,Triton, Ray等框架有基础了解;
3、对于模型训练、推理服务化、推理加速有一定动手经验;
4、对数据卸载,网络通信,计算图等优化有经验者优先;
5、思路清晰,主动性强,有很强的自我驱动力。