职位描述
职位描述
1,利用AI编译器实现GPT3等大模型到大算力芯片架构的mapping(自动并行);
2,设计并开发网页前端,对自动并行的结果可视化展示;
职位要求
1,有一定AI算法与训练基础,了解基础CNN、transformer模型;
2,熟悉常用的深度学习框架中的一种,如pytorch、tensorflow、paddlepaddle、jax;
3,熟悉JavaScript语言,有一定的网页开前端开发基础,了解React、Vue、Angular等框架;
4,编程基础扎实、动手能力强、良好的学习能力、较强的自驱力;
加分项
1,对多机多卡训练的实现有一定了解,对大模型算法与训练有一定了解和基础;
2,对深度编译器如XLA、MLIR、TVM等有一定了解和基础;
3,有过完整的网站搭建经验,对web图表可视化有一定经验。
工作地点
上海市/上海市/徐汇区 西岸国际人工智能中心 云锦路701号