之江实验室 不需要融资

杭州·学术/科研·1000-9999人

深度学习优化工程师

15-25K·13薪

杭州

不限

硕士

全职

职位描述
职位描述
1. 分布式计算/存储/调度、深度学习框架(类TensorFlow/PyTorch)的底层算子优化;
2. 研究业界最新进展,研发图神经网络分布式训练平台,基于软硬件协同加速。

职位要求
1.精通python语言,以及其 Native 接口技术;
2.良好的C/C++编程基础;
3.熟悉机器学习框架如TensorFlow、PyTorch、MXNet等至少其中一种;
4. 熟悉CPU、GPU性能调优的原理及概念(如流水线、缓存、指令宽度等),熟悉Nvidia GPU各系列架构;
有如下经验之一者优先:
1. 熟悉参数服务器、All-Reduce等分布式机器学习框架理念,有实践经验者优先。
2. 了解NCCL通信库,有CUDA、TensorRT加速经验者优先;
3. 熟悉多核多线程优化, 掌握OpenMP, TBB, MKL 中的一项者优先;
4. 熟悉编译原理、虚拟机、引擎优化等技术至少一种的优先;
5. 熟悉GNN算法者优先。
公司介绍
之江实验室成立于2017年9月6日,是浙江省委省政府举全省之力打造的重大科技创新平台,主攻智能感知、人工智能、智能网络、智能计算和智能系统五大科研方向。作为全国首个混合所有制事业单位性质的新型研发机构,实验室已纳入国家实验室体系,获批牵头建设智能科学与技术浙江省实验室,产出了类脑计算机、天枢人工智能开源平台、仿生深海软体机器人等全球领先的科研成果。
2021年5月,之江实验室成立北京研究中心(办公地点位于北京市海淀区中关村),旨在围绕国家重大创新领域,联合在京优势科研机构、高校与大型企业开展关键技术攻关和核心系统研发,承担国家重大战略任务。北京研究中心先期聚焦智能无人系统、专用智能计算系统关键技术研究,打造一支多学科融合、富有创新活力的一流科研团队,争取一批原创性、突破性、引领性的重大科技成果。北京研究中心已与航天五院、中科院、清华大学、中国电子、寒武纪等开展多项科研合作。
杭州余杭区之江实验室1期之江实验室
相似职位