工作职责:

-参与百度百舸分布式训练框架的研发
-参与百度百舸大模型推理引擎的研发
-参与训推引擎中高性能GPU算子的研发
-参与和训推性能有关的模型架构优化

任职要求:

-计算机、人工智能相关方向，硕士及以上在读
-扎实的torch/cuda/tilelang等编程基础，熟悉常用的开源训推引擎
-了解当前主流大模型架构与芯片架构，掌握大模型训推优化的基本方法
-热爱AI技术，有挑战SOTA的意愿和激情