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或芯片特定指令集)项目经验者优先考虑。
牛客安全提示:如发现虚假招聘、广告,或以任何名义索要证件、费用,或诱导异地入职、参与培训等均属违法行为,发现请
立即举报