高性能计算工程师

25-50K·15薪

上海

3-5年

本科

全职

职位描述
岗位职责:
1. 研发高质量的芯片基础软件栈,包括支撑大语言模型在内的深度学习,线性代数,科学计算,信号处理等方向的基础加速库;
2. 分析重要的性能指标,调优已有软件算法,并且对未来的软件进行优化和提升;
3. 和硬件/应用的同事协同工作,一起对大语言模型,CV模型,语音模型,推荐系统,科学计算和其他相关并行算法领域进行分析和优化;
4. 负责 AI模型推理/训练的端到端性能整体优化;
5. 实现AI/科学计算领域在GPGPU上的最新算法和最佳实践;
任职资格:
必须:
1. 熟悉 C++编程,了解常用数据结构及算法;
2. 熟悉线性代数,科学计算,机器学习,深度学习,AI算法等一个或多个领域;
3. 熟悉 GPU 或 AI加速芯片的体系结构,有过在不同架构芯片上的算法调优经验;
4. 熟悉并行算法优化的基本技巧,有对算法性能进行分析和优化的经验;
5. 熟悉计算机体系结构,操作系统;
加分项:
1. GPU的编程和优化经验(e.g. CUDA or OpenCL);
2. 有过机器学习方面的开发经验(尤其是深度学习),熟悉Tensorflow, Pytorch, TensorRT等框架/引擎的原理和使用方法;
3. 有过视觉/大语言模型的开发和调优经验,熟悉CNN/Transformer等网络的性能瓶颈和优化方向,能够把握相关领域的国际前沿发展趋势;
4. 有过数据科学,统计科学,图像处理,信号处理等方面的开发和优化经历;
5. 有软硬件协同优化、基于异构硬件的软件开发经验者优先;
公司介绍
上海天数智芯半导体有限公司(简称“天数智芯”)于2018年正式启动GPGPU大芯片设计,是中国第一家GPGPU高端大芯片及超级算力提供商。

公司以“成为智能社会的赋能者”为使命,专注于云端服务器级的通用高性能计算大芯片,瞄准以云计算、人工智能、数字化转型为代表的数据驱动技术市场,解决核心算力瓶颈问题。重点打造自主可控、国际一流的通用、标准、高性能云端计算GPGPU大芯片,从芯片端解决算力问题;并推出面向5G应用需求的云端推理芯片,与进口主流GPGPU生态兼容,并提供国产化选项,意在将超级计算发展为水电煤一样的基础设施,以更可信、更高效、更经济的方式帮助各行业实现算力升级。
上海浦江智慧广场3号楼5楼
面试经验

匿名用户

面试上海天数智芯的市场营销·上海

面试通过
面试:市场营销。感觉就还好吧,整个过程都没有什么难度,确认通过。
蛮好的
查看更多面经(1)
相似职位