欢迎加入摩尔线程GPU
-Genius团队,我们专注于AI模型工程化,对最热门模型进行训练和推理的适配以及性能优化,探索GPU性能的极限,为全球客户提供先进的解决方案。加入我们,一起探索技术的无限可能!HC很多,欢迎投递!主要职责:
1.PyTorch/Paddle/TensorFlow模型适配与性能调优。
2.大模型分布式训练部署与性能调优。
3.vllm以及推理引擎部署与性能调优。
4.开发高性能算子。
5.对社区热门CUDA仓库进行musify。
6.调研最新的机器学习技术和平台功能。
7.深度参与研发产品特性开发,为研发产品支持新特性。基本要求:● 熟悉GPU架构,熟悉计算机体系结构。● 熟悉CUDA编程,熟悉CUDA kernel性能调优,熟悉c++11新特性。(必须)● 熟悉大模型分布式训练策略。● 至少有一种机器学习框架使用或者开发经验,如TensorFlow、PyTorch、PaddlePadde、Megatron
-LM、Colossal
-ai、DeepSpeed、vLLM或推理引擎。● 良好的逻辑思维能力,精通计算机数据结构和算法;● 熟悉机器学习和深度学习算法及其在实际应用中的部署。● 计算机科学、电子工程或相关领域的本科及以上学历。● 积极乐观,责任心强,工作认真细致,具有良好的团队沟通与协作能力。● 良好的英语读写能力。优先考虑条件:● 2年及以上GPU大模型推理/训练性能优化工作经验。● 算子优化经验● tensor core/wgmma/cutlass/cute经验。微信:
-1001
-14af
-4f71
-979b
-7c6e0b9d297c&language=zh_CN