机器学习框架工程师

岗位职责

1.负责全公司机器学习基础设施的研发，包括研发大规模训练架构、高性能推理服务、GPU硬件加速，提升业务如自动驾驶等场景的模型研发效率；

2.深入分析训练和推理系统，定位和解决系统各个层次的性能瓶颈，不断提升效率。

任职要求

1.计算机、数学、物理、电子工程、自动控制等专业本科及以上学历；

2.熟练掌握python/C++编程语言，具备扎实的代码能力和数据结构基础；

3.熟悉pytorch深度学习框架，了解一般的训练优化技巧；

4.了解常见的深度学习视觉或语言模型；

5.熟悉CUDA kernel研发和优化的优先；

6.熟悉Deepstream等框架者优先；

7.熟悉GPU、RDMA硬件架构，熟练使用CUDA编程的优先。