岗位描述
岗位职责:
1.设计开发机器学习的基础设施、实验网络模型以及优化算子;
2.分析AI业务场景,提炼计算和通信算法模式,调试网络,部署到我司GPU平台,验证功能正确性和执行效率;
3.训练计算机3D图形以及视觉增强方面的AI模型,与GPU 3D渲染管线融合部署;
4.开发网络模型验证所需的各种高效调试工具;
岗位要求:
1.有机器学习方面的经历,比如模型训练、机器学习编译和模型部署等;
2掌握Pytorch框架和Python编程语言, 熟悉软件设计模式;
3.熟悉CNN、Stable diffusion、Transformer 等网络结构;
4.对机器学习技术栈,有浓厚兴趣;
职业发展可能:
1.成为AI应用方面的专家,基于我司GPU,驾驭各种AI应用,引导GPU的迭代演进方向;
2.成为GPU的AI通用计算方面的架构师,比如高效的矩阵乘法器、高效的缓存机制等;
3.成为AI算法专家,尤其在AI通信算法方面;
学历要求:
硕士