logo
更新:2024-05-17
AI 编译器开发工程师
3-6万
上海浦东新区  | 本科  | 社招
去申请
收藏
举报
职位详情
绩效奖金
团建聚餐
员工旅游
节日福利
定期体检
年终奖
五险一金
补充医疗保险
带薪年假
意外险
生日福利
职位描述:
我们正在寻找一位AI 编译器开发工程师,具有TVM、MLIR、XLA等编译器技术开发工具链经验,熟悉PyTorch、TensorFlow,ONNX,熟悉C++和Python等编程语言,来加入我们的团队。在这个角色中,您将负责使用这些工具和技术构建和优化AI编译器工具链,并能够在我们的硬件架构的生产环境中自动调整以提高AI模型的性能。
主要职责:
1.利用开源工具和框架技术,构建和优化自研AI编译器工具链,包括深度学习模型的编译、优化和部署
2.分析深度学习网络,开发和实施编译器优化算法
3.参与芯片软硬件协同开发
4.对深度学习框架进行性能调试和优化
5.与团队合作,开发和优化编译器基础设施,以提高AI模型的性能
6.设计和实现新的编译器优化技术
7.提供技术支持,协助团队解决AI模型编译和部署中的技术问题
职位要求:
1.计算机科学,机器学习,数学,物理,电子工程或相关领域的硕士或以上学位,或相关领域的实际工作经验。
2.具有3年以上的编译器、解释器或相关工具链的开发经验。
3.熟悉TVM、MLIR、XLA等编译器基础设施,具有使用这些工具进行AI模型编译和优化的经验。
4.熟悉深度学习模型和算法,具有使用PyTorch、TensorFlow深度学习框架和适配ONNX / Pytorch / Tensorflow推理框架的经验
5.熟悉AI训练框架及其适配
6.熟悉C/C++和Python等编程语言。
7.具有优秀的团队合作和沟通能力,能够在团队中提供技术支持和指导。
8.能够快速学习新技术,适应快速变化的工作环境。
9.对AI编译技术和计算机体系结构有浓厚的兴趣,愿意投入时间和精力进行深入研究。
优先考虑有以下一条或多条经历:
1.具有使用TVM / MLIR / XLA经验进行AI模型编译和优化的经验
2.自研GPGPU / NPU 等AI工具链经历
3.具有在开源社区贡献TVM/MLIR等编译器项目的经验
4.具有使用PyTorch、TensorFlow、Caffe或其他深度学习框架进行模型训练和部署的经验。
工作地址
上海浦东新区WeWork(浦东嘉里城)4楼
公司介绍
博瑞晶芯是一家主要从事处理器芯片的设计与销售,由财务投资人、产业与战略投资人等出资成立的商业市场化初创型企业。公司塑造开放包容的企业文化与务实进取的管理风格,总部设立在深圳,同时将在上海、北京、成都及海外设立研发中心。创始团队与核心研发人员来自于国际与国内的知名处理器设计企业,平均20年以上的CPU核与处理器SoC设计、验证及商业化经验。公司主要基于Arm架构开发服务器SoC产品,并通过灵活的合作模式以满足数字化转型中各行业对处理器的共性或差异性需求,主要目标客户为互联网企业、运营商及垂直行业。
若用人单位提供虚假招聘信息,以担保或其他任何名义收取财物,扣押或以保管为名索要证件,都属于违法行为,应当提高警惕。
发布于boss直聘