一、崗位職責(zé)
1.負(fù)責(zé)軌道交通場景CV大模型建設(shè)與應(yīng)用;
2.負(fù)責(zé)CV大模型的訓(xùn)練、微調(diào)、驗證、部署與集成,確保模型達(dá)到預(yù)定性能目標(biāo);
3.負(fù)責(zé)推進(jìn)分布式訓(xùn)練中的并行計算、通信優(yōu)化、穩(wěn)定性提升等工作;
4.負(fù)責(zé)利用多 GPU、多節(jié)點(diǎn)進(jìn)行模型訓(xùn)練,提高訓(xùn)練效率,來提高模型的泛化能力;
5.組織與相關(guān)業(yè)務(wù)進(jìn)行需求對接,撰寫技術(shù)報告、模型 / 算法文檔等。
二、任職要求
1.計算機(jī)科學(xué)或人工智能相關(guān)專業(yè),碩士(5年以上工作經(jīng)驗)、博士;
2.熟悉模型部署流程,能夠在不同的生產(chǎn)環(huán)境中部署模型,如云平臺、邊緣設(shè)備等,實(shí)施模型監(jiān)控策略,跟蹤模型性能,確保模型在生產(chǎn)環(huán)境中的穩(wěn)定性和可靠性;
3.具有獨(dú)立開發(fā)能力以及合作開發(fā)能力,具有較強(qiáng)的團(tuán)隊精神、良好的溝通協(xié)調(diào)能力、較強(qiáng)的分析和解決問題的能力。
三、專業(yè)技能
1.優(yōu)秀的代碼能力、數(shù)據(jù)結(jié)構(gòu)和基礎(chǔ)算法功底,熟悉CV相關(guān)開源模型、機(jī)器學(xué)習(xí)和深度學(xué)習(xí)模型(CNN、LSTM、 Transformer、GPT等),熟練掌握Python,熟悉Linux開發(fā)環(huán)境;
2.具備視覺大模型訓(xùn)練、調(diào)優(yōu)和評估能力;
3.有機(jī)器學(xué)習(xí)框架經(jīng)驗者優(yōu)先(TensorFlow/PyTorch);
4.有分布式訓(xùn)練工具經(jīng)驗者優(yōu)先(Colossal-AI/DeepSpeed/Megatron)。