AI芯片算子开发工程师(北京/深圳)

岗位职责

1.基于自研AI芯片,研发高性能算子、对算子和模型进行调试调优;

2.针对业务落地过程中的性能与泛化性问题,不断迭代算子库;

3.能够编写高效的计算核和数据流,交付性能极致的算子;

4.不断迭代算子工具链,以确保其完整性和易用性,和业务一起构建AI生态。

岗位要求

1.熟悉Linux开发环境,掌握Python/C++等语言, 具备良好的编程基础以及系统设计能力;

2.能够使用nvprof、nsys等profile工具分析算子性能,制定优化方案并高效实现;

3.熟悉GPU或SIMD编程体系结构和原理,对Cuda、ROCm算子具备一定的开发能力;

4.熟悉Pytorch、Tensorflow框架和自定义算子开发,熟悉LLM、AIGC等常见模型结构;

5.精通常用的算子优化手段,如剪枝、量化、分布式推理等技术;

6.具备良好的技术热情和责任感、优秀的分析解决问题能力、良好的团队意识和沟通能力。

加分项

1.有线性代数和概率论基础,熟悉数学建模、算法设计和优化方法者优先;

2.有FasterTransformer、TensorRT等使用和分析经验者优先。

公司地点:深圳南山区深圳金地威新中心1层

公司简介:

腾讯以技术丰富互联网用户的生活。

通过通信及社交平台微信和 QQ 促进用户联系,并助其连接数字内容和生活服务,尽在弹指间。

通过高效广告平台,协助品牌和市场营销者触达数以亿计的中国消费者。

通过金融科技及企业服务,促进合作伙伴业务发展,助力实现数字化升级。

我们大力投资于人才队伍和推动科技创新,积极参与互联网行业协同发展。

腾讯于 1998 年11月在中国深圳成立,2004 年6月在香港联合交易所主板上市。

职位发布者:周女士

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

融资阶段:

公司规模:1000~9999人

相似职位: