30-60K·14薪

AI芯片算子开发工程师

深圳3-5年本科

分布式训练 算子开发 Python
立即沟通
王先生
感兴趣

腾讯 ·招聘经理

职位描述

岗位职责:
基于自研芯片,研发高性能算子,算子和模型调优,构建算子算法Demo,解决开发者问题;
针对业务落地过程中性能、泛化性问题,不断迭代算子库性能和泛化性;
迭代优化算子工具链的完备性和易用性,和业务一起构建AI生态;
岗位要求:
熟悉Linux开发环境,掌握Python/C++等语言, 有良好的编程基础、系统设计优化能力;有线性代数和概率论基础,熟悉数学建模、算法设计和优化方法;
熟悉GPU/AI DSA体系结构和原理,芯片开发优化经验,精通Cuda、ROCm算子开发;
熟悉Pytorch/Tensorflow框架和自定义算子开发,熟悉LLM、AIGC等常见模型结构;有FasterTransformer、TensorRT等使用和分析经验者优先。精通常用的算子优化手段,如剪枝、量化、分布式推理等技术;
能够使用nvprof/nsys等profile工具分析算子性能,制定优化方案并高效实现;
了解SIMD架构AI芯片或CPU向量指令的算子研发经验,能够编写高效的计算核和数据流,交付性能极致的算子;
具备良好的技术热情和责任感,优秀的分析解决问题能力,良好的团队意识和沟通能力。

公司介绍

腾讯以技术丰富互联网用户的生活。
通过通信及社交平台微信和 QQ 促进用户联系,并助其连接数字内容和生活服务,尽在弹指间。
通过高效广告平台,协助品牌和市场营销者触达数以亿计的中国消费者。
通过金融科技及企业服务,促进合作伙伴业务发展,助力实现数字化升级。
我们大力投资于人才队伍和推动科技创新,积极参与互联网行业协同发展。
腾讯于 1998 年11月在中国深圳成立,2004 年6月在香港联合交易所主板上市。

工商信息

腾讯科技(上海)有限公司

法定代表人:奚丹 注册资本:500万美元
成立日期:2008-07-23 经营状态:存续

工作地址

深圳南山区深圳金地威新中心B栋B栋
腾讯

腾讯科技(上海)有限公司

互联网不需要融资 10000人以上

更新于:2024-11-05
下载APP确定

已安装?在APP中 直接打开