Cuda工程师
|
|
宁德市 |
|
2024-05-06 |
Seniority level职位级别 | 中高级 |
Industry公司行业 | ^汽车零配件[汽车零配件]^ |
Employment type职位性质 | 全职 |
Function工作职能 | |
Number招聘人数 | 1 |
Responsibilities(职位描述)
岗位职责:
1.负责工业领域AI软算框架的开发;
2.实现或优化大模型框架并行策略(如Megatron)提升大模型训练性能;
3.优化或重写计算框架的算子以达到最优的性能;
4.分析、定位、解决模型训练或推理过程中涉及的计算框架问题。
Requirements(任职要求)
岗位要求:
1.具有计算机相关领域背景,熟悉AI领域基本知识;
2.熟练使用Python/C++编程,熟悉CUDA编程、熟悉openCV;
4.熟悉PyTorch、PaddlePaddle、Megatron等主流框架;
5.熟悉DP、TP、PP等大模型并行策略,熟悉并行库MPI、NCCL;
6.熟悉Transformer、GPT主流模型结构,熟悉NLP常用算法原理;
7.有AI框架、算子开发经验者优先;