AI工具链开发实习生(模型部署方向)

200-500元/天
算法工程师
北京/深圳
硕士
5天/周
最少3个月
有转正
岗位关键词
岗位职责
1. 负责开发面向自研芯片的AI模型编译工具链,支持CNN、Transformer等主流网络的高效部署。 2. 设计并实现模型优化、图编译、算子融合、内存优化等关键技术,提升芯片上的推理性能。 3. 开发自动化工具链,简化模型从训练框架(如PyTorch/TensorFlow)到自研芯片的部署流程。 4. 与芯片团队紧密合作,针对硬件特性(如指令集、内存架构)进行深度优化。 5. 负责工具链的API设计、文档建设和开发者体验优化,推动工具链在客户侧的落地。
岗位要求
1、计算机、电子工程、数学等相关专业硕士及以上学历; 2、熟练掌握Linux环境下Python和C++开发,同时具备高性能代码开发和调试能力者优先; 3、熟练使用Git, CMake, Docker, SSH等常用开发工具; 4、了解AI模型部署全流程,包括模型量化、剪枝、编译优化等; 5、了解CNN/Transformer等网络结构及其计算特性; 6、熟悉至少一种主流深度学习框架(PyTorch/TensorFlow/MXNet等); 7、有异构计算性能优化(如CUDA/OpenCL或芯片特定指令集)项目经验者优先考虑。
牛客安全提示:如发现虚假招聘、广告,或以任何名义索要证件、费用,或诱导异地入职、参与培训等均属违法行为,发现请 立即举报