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