多元算力AI计算引擎工程师/专家
🤖 AI 估测：¥45K-80K
发布时间：16 天前
ℹ️关于这个职位
这是一个专注于AI大模型底层计算系统的核心研发岗位
你将负责设计和实现大模型的强化学习、SFT/蒸馏训练框架，以及分布式推理引擎，目标是提升大规模集群的训练性能和推理效率
同时，你需要评估和适配多种异构计算芯片（如GPU/NPU），并开发相应的计算加速引擎
✓工作职责
跟踪业界最新强化学习领域算法/训练范式进展，完成大模型强化学习训练框架的设计与实现，提升大集群分布式训练性能，缩短模型迭代周期
负责大模型SFT/蒸馏训练框架的设计与实现，提升分布式训练性能
负责分布式大模型推理引擎的方案设计与实现，快速完成业界Sota模型的规模化上线服务
负责异构计算芯片（GPU/NPU/ASIC等）的评估、选型以及计算加速引擎设计实现
⭐最低要求
深入理解深度学习框架底层实现（如TensorFlow/PyTorch的计算图优化、运行时调度等）
熟悉至少一种主流异构计算架构（如NVIDIA CUDA、AMD ROCm、Google TPU等）及其编程模型
具备优秀的算法思维、系统架构设计能力与工程实现能力
对AI技术前沿有强烈兴趣，能快速学习并解决复杂技术问题
👍优先资格
具备大集群环境下进行大模型训练开发经验者优先
具备高性能推理引擎开发经验者优先
具备高性能算子库、通信库开发、优化经验者优先