工作职责:
-参与百度百舸分布式训练框架的研发
-参与百度百舸大模型推理引擎的研发
-参与训推引擎中高性能GPU算子的研发
-参与和训推性能有关的模型架构优化
任职资格:
-计算机、人工智能相关方向,硕士及以上在读
-扎实的torch/cuda/tilelang等编程基础,熟悉常用的开源训推引擎
-了解当前主流大模型架构与芯片架构,掌握大模型训推优化的基本方法
-热爱AI技术,有挑战SOTA的意愿和激情