Ivyea Jobs 6511 roles · 88 companies · 刚刚
智元机器人 · 具身智能

AI编译器开发实习生

上海 / 北京 实习 推理 / 部署优化

职位描述

全栈开发自研编译器,辅助算法业务落地 跟进算法交付,在规定时间内交付可部署的算法,并支持多芯片部署 持续优化编译器各层能力,保持编译效果对齐业界SOTA

任职要求

独立部署过vla,vlm类型模型; 理解AI框架及常见的AI模型; 熟练掌握python以及C/C++编程; 对AI编译技术栈有深度理解,前端、优化、后端有基本认知 熟悉transformers,hugginface,pytorch等框架的使用并了解算法搭建过程 深度理解torch.compile, torch.trace, jax.jit过程原理,有能力进行AST语法树解析 熟练使用TVM,TensorRT等框架并有实际部署经验 了解算子优化与集成常用技术,可以使用算子注册加速模型
智元机器人去官方页面投递 →