AI for Science 高性能计算研发工程师

【岗位职责】 你将参与曦望自研 GPU 平台的科学计算与 AI4Science 软件生态建设，面向科学模型和数值仿真中的实际计算问题，开展框架适配、数值算法实现及性能优化，推动 AI 模型与传统科学计算在国产平台上协同运行。 面向气象与流体、分子动力学、蛋白质与材料等科学计算场景，分析模型和数值算法的计算特征，完成代表性应用的迁移、验证与端到端加速。 参与 JAX／OpenXLA、PyTorch 等框架在自研 GPU 上的适配，开发所需的编译、运行时及自定义算子支持，完善自动微分、批量计算和多设备执行能力。 开发和优化 FFT、稠密与稀疏线性代数、迭代求解、Stencil、邻域搜索、Scatter／Gather 及归约等核心计算模块，构建可复用的高性能算子与数值计算库。 面向 FNO 等神经算子、图神经网络、等变网络、分子机器学习势及科学基础模型，开展模型结构与硬件协同优化，提升训练和推理…… 完整岗位描述与官方投递入口,请登录后在岗位详情页查看。

本站当前在招 21411 个北京岗位（全城各职能合计），其中 2446 个公开标注了薪资。按这些标注统计：月薪（税前）主流区间 5.5K ~ 19K（25~75 分位），中位约 9K，最低 1K、最高 300K。