职位&公司对比
职位详情
- 深圳
- 10年以上
- 硕士
- C/C++
- 网络结构搜索
- 深度学习
- 软硬件联合优化
- 并行计算
- 混合精度
- 模型加速/性能优化
- Python
工作职责 1. 负责芯片SDK的架构设计和算法优化： a) 跨量化、DSP部署和NPU部署的NNSDK，性能/效率/质量持续对标业界优秀芯片SDK； b) 联合网络结构、量化和编译的优化算法设计和规划，效果持续对标业界SOTA； 2. 负责DSA架构NPU/DSP编译器工具链设计、开发和维护，实现对TensorFlow/PyTorch/Onnx等主流深度学习框架的支持； 3. 基于TVM/MLIR等开源框架做二次开发，实现编译器工具链中各种优化算法； 4. 开发AI算法模型高效算子，负责NN算法、图像算法在嵌入式异构平boss台上的实现； 5. 通过与相关领域（算法、芯片）的联合设计提升计算性能，并设计下一代AI编译器架构规范。 任职要求 1. 硕士及以上学历，具备计算机科学、信息工程、电子工程、自动化等相关专业背景； 2. 具备2年以上异构计算平台算法优化经验，对以下一种或多种异构计算硬件体系结构、并行计算有深入研究和实践经验： a) 基于业界前沿AI加速器平台上算法性能优化经验； b) 基于业界前沿GPU平台的算法性能优化经验，熟悉GPU架构编程； c) ARM/x86等CPU上算法性能优化经验，熟悉多线程编程和内存管理技术； 3. 具备3年以上AI编译器框架开发经验： a) 熟悉TensorFlow/PyTorch/BOSS直聘Onnx等主流深度学习框架； b) 有基于TVM/MLIR等开源编译器开发经验； c) 有NPU/DSP/GPU等异构计算平台开发和调优经验； 4. 熟悉C/C++以及python等编程语言，熟悉数据来自BOSS直聘结构和软件直聘设计模式，具备良好的软件工程习惯； 5. 具备良好的沟通协kanzhun调和组织能力、技术项目管理能力。
职位详情
- 深圳
- 3-5年
- 本科
- 大模型算法
- 模型加速/性能优化
- C/C++
- Python
岗位描述：负责模型训练框架和工具的开发、优化等, 提高模型训练的吞吐； 1.模型训练框架的优化 a 提升模型训练框架训练的吞吐, 上手的易用程度； b 提升模型训练的效率, 包含不仅限于数据流的优化(NVIDIA DALI), 模型(forward & backward)的优化, (包含不仅限于高效算子编写替换, torch compiler), amp使用(accelerate)； c 模型训练精度提升, amp/int8 aware training； d 模型训练框架的重构, 推进文档化, 易用性； 2.模型训练可观测性的提升 a 建立模型训练客观性指标, 基于torch.profile或者其他打点工具, 进行多层次粒度上的观测性；、 b 基于现有的模型, 探索模型训练的linear scaling raw； 岗位要求： 1.计算机相关专业本科以上学历 ；优秀的编程能力和数据基础，熟练使用C++/Python； 2.熟boss悉pytorch底层, 完整了解训练pipeline, 具备数据流优化, kanzhun训练优化经验 a 熟悉accelerabosste, megatron等开源的分布式训练框架优先 b 了解训boss练优化技巧, 包含不限于torch.compiler/算子编写 c 熟悉openai triton 3.热衷于优化end-to-end training system、理解高性能I/O, 用以最大化本地性能以及分布式性能资源
技能解析
- 深度学习框架
- 项目管理
- 编程语言
- 架构设计
- 软件工程
- 组织能力
- 算法优化
- 内存管理
- C/C++
- 熟悉数据
- 线程编程
- 性能优化
- 开发经验
- 算法设计
- 开源框架
- 熟悉多线程
- 性能优化经验
- 图像算法
- 项目管理能力
- 优化算法
- 深度学习
- 软件设计
- 沟通协调
- 多线程编程
- 电子工程
- 数据结构
- 设计模式
- 二次开发
- 熟悉数据结构
- 管理能力
- 框架开发
- 好的沟通
- 算法模型
- 良好的沟通协
数据来自CSL职业科学研究室
技能解析
- 训练框架
- 模型训练
- 编程能力
- 熟练使用C
数据来自CSL职业科学研究室
公司福利
- 交通补助
- 生日福利
- 节日福利
- 零食下午茶
- 餐补
- 员工旅游
- 带薪年假
- 股票期权
- 年终奖
- 定期体检
- 五险一金
- 团建
- 弹性时间
- 企业滴滴
- 加班餐补
公司福利
- 节日福利
- 包吃
- 带薪年假
- 年终奖
- 定期体检
- 五险一金
备注
职位发布者未明确表明公司信息，具体可咨询职位发布人进行确认。
备注
职位发布者未明确表明公司信息，具体可咨询职位发布人进行确认。
电子营业执照
京公网安备11010502032801