崗位描述:
模型工程師負責部署、訓練和優化AI模型,特別是針對圖像生成任務(如Stable Diffusion)。負責確保模型在高效能的硬件環境中運行,并進行調優,確保其生成的內容符合項目需求。
主要職責:
部署和訓練AI模型,尤其是圖像生成類的模型(例如Stable Diffusion、Lora)。
優化和調優訓練過程,提升生成效果和性能。
配置與維護AI模型運行環境(如CUDA、Python環境等)。
協助團隊解決技術問題,提供技術支持
基礎職責說明:
負責搭建、訓練和優化AI生成模型。
調優生成模型以滿足業務需求,包括定制化模型的微調、訓練流程的優化。
基礎要求說明:
· 基本技能要求:具備深度學習框架(如PyTorch、TensorFlow)和生成模型(如Stable Diffusion、GPT等)的應用經驗。
· 編程能力:精通Python和至少一個深度學習框架,能夠進行模型訓練、調優和部署。
· 算法理解:具備扎實的機器學習/深度學習理論基礎,能獨立進行模型調優。