AI算法工程师

职位描述

职位描述

1、结合算法发展趋势，提出芯片架构优化建议，完善硬件架构

2、支撑大模型推理及训练业务的运行加速与优化

3、将vLLM、SGlang、Megatron等深度学习推理和训练框架适配到公司自研的芯片上；

4、参与AI agent相关构建、训练

5、前沿技术探索：跟踪大模型架构、多模态学习、强化学习、agent等前沿方向

任职要求

职位要求

1、计算机科学、电子工程或相关领域的硕士学位；

2、熟悉C/C++/Python软件编程语言；

3、了解深度学习、计算机视觉的基础知识，熟悉CV, LLM, VLM等模型的原理；

4、具备一定的PyTorch框架使用经验，了解vLLM、SGlang和Megatron等框架在大语言模型中的应用和优化；

5、了解Prompt工程、agent、RAG等大模型应用算法架构和相关技术

6、优秀的分析问题和解决问题的能力，良好的团队沟通能力、技术攻关能力，责任心强，具备一定抗压能力。

7、了解LangGraph等框架，有agent RL、Lora训练经验优先