AI 內核工程師

我們正在尋找對跨平台性能優化與軟硬體協同設計充滿熱情的 AI Kernel 工程師。你將研究並整合多家 GPU 廠商 (NVIDIA CUDA、AMD ROCm、Intel oneAPI) 的架構與運算庫，並主導核心 AI 演算法 (如 Transformer Engine、FlashAttention) 的跨平台適配與最佳化。我們也重視具備 AI 特化加速器 (如 AWS Trainium、Google TPU 及其他 ASIC) 軟體開發經驗的工程師，致力於推動跨異質計算平台的效能一致性與整合。

工作內容

• 研究並分析各家 GPU 架構與加速庫 (cuBLAS、rocBLAS、oneDNN 等) 的設計原理與效能特性
• 整合並優化 Transformer Engine、FlashAttention 等核心算法，確保跨平台效能
• 主導 Kernel 跨平台移植、效能調優與庫整合自動化
• 研究並整合 AI 加速晶片 (如 Trainium、TPU、自訂 ASIC) 於訓練與推論流程中的應用
• 與硬體/軟體供應商協作，導入與整合最新技術與功能
• 建置與維護跨硬體平台的一致性內部運算核心庫，並持續最佳化效能

必備條件

• 具備 CUDA、ROCm 或 oneAPI 開發經驗
• 具備 Kernel 效能調校、跨平台移植與優化實績
• 具備 AI 特化加速器 (如 Trainium、TPU 或其他 ASIC) 的開發經驗
• 深刻理解核心 AI 運算算法 (如 GEMM、Attention、Transformer 核心)
• 具備獨立研究、系統分析與提出架構解決方案的能力
• 熱衷於追蹤最新 AI 基礎設施領域的軟硬體技術發展

加分條件

• 有與硬體/軟體供應商合作或參與硬體協同設計專案經驗
• 熟悉 Kernel 效能測試自動化、CI/CD 與效能驗證流程
• 了解異質運算架構與統一程式設計模型

請將履歷寄至 contact@unieai.com。