企业招聘
职位
关于禾蛙
合伙人计划
联系客服快速发单
扫码添加企业微信
客服电话
400-7777-5125
猎企入驻
400-777-5125
免费注册
登录
**********************
高性能计算优化专家
接单
收藏职位
分享职位
50-100万
北京
研究生
不限
招聘 3 人
预计佣金
140K
三周前发布
JD基本信息
岗位职责
职位描述 1. 主导科学智能领域核心项目的高性能优化工作,重点突破OpenLAM系统级性能瓶颈,包括: - 面向上亿级科学数据的大模型训练场景,优化分布式训练框架的通信效率与计算资源利用率; - 软硬结合优化算子效率,极限压榨硬件性能; - 探索CUDA内核级优化与PyTorch计算图编译技术的前沿应用; - 构建超大规模晶体材料数据库(亿级原子体系)的高效数据预处理与并行计算方案; 2. 深度参与AI for Science基础设施体系建设: - 将优化成果沉淀为通用加速组件,持续提升DeePMD-kit等开源软件的计算效率; - 构建面向材料、能源等领域的端到端加速方案,驱动工业级科学软件的研发。
任职要求
职位要求 【必要项】 - 计算机/应用数学/计算材料学等相关专业硕士及以上学历(非常优秀的本科生也可考虑),2年以上HPC优化经验 - 精通CUDA编程与GPU性能分析工具(Nsight/VTune),有PyTorch/TensorFlow框架层优化经验 - 掌握MPI/OpenMP并行编程技术,具有千卡级分布式训练实战经验者优先 - 熟悉典型AI模型计算特征(Attention/卷积/稀疏计算),能针对性设计内存/通信优化方案【加分项】 - 有MLSys/SC/PPoPP等体系结构顶会论文 - 熟悉科学计算典型负载(DFT/MD等)的加速技巧,或参与过Deep Modeling社区其他明星项目或AlphaFold等其他知名科学智能项目 - 具备物理化学场景计算优化经验,或超大规模图数据处理实战经历成长地图 - 学术纵深:顶会/期刊论文发表,与北京大学/普林斯顿等顶尖实验室深度合作 - 产业落地:对接知名材料、制药产业伙伴,见证技术成果转化为工业软件 - 职业弹性:获得AI算法研发与材料计算(院士团队)双重培养路径
所属行业:
人工智能AI
职能分类:
C++
工作城市:
北京,招聘3人,详细地址:北京海淀区智源大厦
职位要求
学历要求:
研究生·统招·985/211
工作年限:
不限
技能/证书:
-
薪资福利
年薪范围:
50-100万*15薪
薪资福利:
-
团队架构
所属部门:
-
下属人数:
-
部门架构:
-
汇报对象:
-
职级职称:
-
面试信息
面试轮次:
-
面试流程:
-
视频面试:
不可以接受
举报