工作职责:
该职位负责管理燧原框架组的日常工作和人事。和团队架构师制定团队的技术产品路线;和PL,PM一起合作交付项目目标。具体包括以下几项:
带领团队负责第三方框架:Tensorflow/PyTorch/ONNX/MindSpore/Paddle 等的对接方案和技术支持。扩展燧原芯片的框架支持生态。
带领团队分析和优化AI框架的性能,添加,维护框架层的优化pass(es)
带领团队负责AI框架AOT/JIT 编译功能支持,动态图和静态图的支持
带领团队针对神经网络计算设备,设计和调优AI框架图级别调度和融合等算法优化;
跟踪/分析/评估各主流开源深度学习框架
参与自研AI芯片系统软件栈的整体设计和实现,性能调优
和上下左右团队协同合作,拉通对齐需求和风险
职位要求:
o 计算机/电子信息类专业本科,硕士及以上学历
o 理解深度学习理论,熟悉开源AI框架的源码和技术特点:Tensorflow/PyTorch/ONNX/MindSpore/Paddle etc.
o 熟悉MLIR/LLVM, TVM 等AI编译栈技术。
o 在知名公司负责过框架组的技术工作和产品。
o 有扎实的C/C++/Python编程能力
o 对计算机体系结构有深刻理解
o 有较强的快速学习能力,良好的团队合作能力和沟通能力
o 流利的英语阅读、写作能力
o (加分项)有编译器(LLVM/GCC),MLIR框架经验
o (加分项)有CUDA, TensorRT, OpenCL或其他并行计算/HPC开发经验
o (加分项)有XLA,TVM,ONNX, PlaidML, Halide, IREE, Tiramisu and Glow计算图优化经验
o (加分项)有分布式计算框架的研发、运维、性能评测调优等经验
o (加分项)有在基于深度学习的目标识别、目标检测、机器视觉、NLP领域有过工程经验或发表过学术论文"
欢迎咨询!
电话:18519274080
微信号:Brylin1991
邮箱: herocanjob@163.com
--
FROM 122.190.149.*