崗位職責(zé):
1.負(fù)責(zé)大小模型國產(chǎn)芯片遷移;
2.開發(fā)算子以及遷移工具,保證遷移的準(zhǔn)確性和推理/訓(xùn)練效率;
3.開發(fā)推理框架,兼容多種模型格式和多種算力芯片。
任職要求:
教育背景:計(jì)算機(jī)、電子、自動(dòng)化和數(shù)據(jù)科學(xué)等相關(guān)專業(yè),碩士以上學(xué)歷。
知識(shí)技能:
1.熟悉英偉達(dá)CUDA算子開發(fā)以及其主流SDK生態(tài);
2.有主流模型比Deepseek、Qwen的模型的CUDA算子遷移經(jīng)驗(yàn),完成端到端精度對齊和效率優(yōu)化。
工作經(jīng)驗(yàn):
1.2年以上國產(chǎn)算力芯片算子開發(fā)經(jīng)歷;
2.具備模型開發(fā)、訓(xùn)練和調(diào)優(yōu)及前沿訓(xùn)練與推理技術(shù)等方面相關(guān)工作經(jīng)驗(yàn)。
能力素質(zhì):具備較強(qiáng)的學(xué)習(xí)能力、團(tuán)隊(duì)協(xié)作能力、溝通能力、責(zé)任意識(shí)及上進(jìn)心。