一、職位描述
負責基于深度學習（神經(jīng)網(wǎng)絡、神經(jīng)算子）的智能 PDE 求解算法庫的開發(fā)，專注于智能 PDE 求解庫的重構與集成，提升其模塊化、可擴展性和計算效率，提供穩(wěn)定、可擴展的 API 和計算引擎。
該崗位主要涉及：
1、重構現(xiàn)有算法框架，優(yōu)化代碼結(jié)構，提高可讀性、復用性和擴展性。
2、集成不同的深度學習方法，實現(xiàn)高效、可復用的 API 設計。
3、優(yōu)化求解庫的工程化能力，支持高性能計算、并行訓練和跨平臺部署。

二、主要職責
1.算法庫重構：
o使用面向?qū)ο笤O計（OOD）和設計模式，提高框架可擴展性。
o規(guī)范 API 設計，提供清晰的接口，確保與前端、后端無縫集成。
2.算法集成：
o統(tǒng)一并封裝基于深度學習（神經(jīng)網(wǎng)絡、神經(jīng)算子）的不同 PDE 計算方法。
o設計自適應求解策略，支持自動選擇*優(yōu)算法。
3.高性能優(yōu)化：
o使用 CUDA 進行 GPU 加速，提升計算效率。
o支持分布式訓練（如 PyTorch Distributed / Ray / MPI），優(yōu)化大規(guī)模 PDE 計算。
4.工程化 & 可視化：
o與前端人員協(xié)作，實現(xiàn)求解結(jié)果可視化、實時交互，提供直觀的實驗反饋。
o撰寫技術文檔，支持開源項目維護，推動社區(qū)交流。

三、任職要求
1.計算機、數(shù)學、物理、計算機科學等相關專業(yè)，本科及以上學歷。
2.熟練掌握 Python，具備高質(zhì)量代碼編寫能力，有大型項目開發(fā)經(jīng)驗者優(yōu)先。
3.熟悉 PyTorch / NumPy 等深度學習和科學計算庫，具備深度學習算法工程經(jīng)驗。
4.具備良好的軟件架構設計能力，有深度學習算法框架二次開發(fā)經(jīng)驗者優(yōu)先。
5.熟悉 CUDA / JIT 編譯優(yōu)化者優(yōu)先，有分布式訓練開發(fā)經(jīng)驗者更佳。
6.具備 API 設計、代碼封裝、自動化測試經(jīng)驗，能提升算法庫的工程化能力。

四、加分項
熟悉PDE智能求解和自動微分框架。
具備分布式訓練（如 Ray / Horovod） 經(jīng)驗，優(yōu)化大規(guī)模求解。

五、協(xié)議簽訂
1.協(xié)議期：三年期
2.協(xié)議類型：勞務協(xié)議
3.工作地點：杭州或北京
4.試用期：三個月，試用期間工資80%
5.在本協(xié)議生效期限內(nèi)，試用期限外申請離職者，需承擔相應違約金