職位描述
崗位職責:1、參與國產芯片接入深度學習訓練與推理框架的開發工作2、運用編譯、融合算子等多種技術手段,加速國產芯片上的大模型推理3、其他相關前沿技術的調研與運用崗位要求:1、具有良好的編程習慣,熟練使用Python/C++ 語言,有較強的debug能力;2、熟悉大模型推理算法,熟悉一種或者多種推理框架(vllm, lmdeploy, sglang, lightllm等)3、對深度學習框架有較深理解與實踐(包含不限于深度學習框架:Pytorch,TF,JAX)4、較強的自驅動力,對先進技術有濃厚興趣;5、有較強的團隊精神和溝通交流能力;加分項1、熟悉英偉達或者國產硬件的編程模式2、對深度學習編譯器有較深理解以及擁有實際場景中的應用經驗(torch.compile, triton,XLA,MLIR,TVM,Mojo等等)
企業介紹
上海人工智能實驗室作為新型研發機構,于2020年7月在世界人工智能大會正式揭牌。實驗室面向上海科創中心建設需求,開展戰略性、原創性、前瞻性的科學研究與技術攻關,突破人工智能的重要基礎理論和關鍵核心技術,支撐我國人工智能產業實現跨越式發展,目標建成國際一流的人工智能實驗室,成為享譽全球的人工智能原創理論和技術的策源地。牽頭科學家為湯曉鷗、姚期智、陳杰等人工智能領域世界知名學者。