logo
更新:2024-05-06
算子开发工程师
3-6万
苏州虎丘区  | 5-10年  | 硕士  | 社招
去申请
收藏
举报
职位详情
岗位职责:

1.利用深度学习编译技术,结合专家级经验(算法/系统/硬件),解决搜广推、CV、NLP,以及多模态、大模型的训练和推理场景挑战。

2.系统化分析模型,提炼/抽象优化方法,寻求解空间(包括算子调度,融合,Schedule,异构调度等)。

3. multi-framework(例如Tensorflow、Pytorch等)计算图优化Pass开发,手段包括不限于编译 IR优化,专家经验模板化等等。

4. multi-device(例如GPU,CPU,NPU)专项 SIMT、SIMD 计算加速,包括核心Gemm 算子,Reduce算子,以及复杂融合算子等等。岗位基本要求:

有扎实的计算机理论基础,熟练掌握C++和 Python语言具备以下优先:

1.熟悉PyTorch 或TensorFlow等引擎的架构和运行原理,有深度定制经验优先

2.熟悉TVM/MLIR/XLA等DL编译器,有Cutlass技术经验,具备计算图翻译至IR并优化的经验优先

3.在深度学习领域有丰富的模型优化经验,熟悉 GPU或NPU架构,有CUDA或NPU编程经验优先

4.学习能力和好奇心强,具备良好的问题分析和解决能力,有团队合作精神和沟通能力

其他信息

行业要求:全部行业
工作地址
苏州-虎丘区佳兆业悦峰大厦
公司介绍
亿铸科技成立于2020年6月,致力于用存算一体架构设计AI大算力芯片,将新型存储器ReRAM和存算一体架构相结合,通过全数字化的芯片设计思路,在当前产业格局的基础上,提供一条更具性价比、更高能效比、更大算力发展空间的AI大算力芯片换道发展新路径。
亿铸科技拥有非常优秀的研发、工程及顾问团队。研发能力覆盖了工艺器件、架构设计、电路设计和软件生态等全链条;工程团队核心成员平均拥有25年以上的高端集成电路设计和量产经验,有着丰富的应用和产品化实战经历。
若用人单位提供虚假招聘信息,以担保或其他任何名义收取财物,扣押或以保管为名索要证件,都属于违法行为,应当提高警惕。
发布于猎聘网