深度学习框架工程师
面议 | 硕士及以上
计算机类 | 招聘10人 | 北京市市辖区 | 中国信息安全研究院有限公司
来源:国聘
职位职责:
【岗位职责】 1.负责面向具体任务的深度学习训练/推理性能优化。优化方向侧重于GPU多线程计算、访存方面 2.负责分析优化当前基于Transformer的大模型(如SwinTrasformer,LLaMA等)的性能瓶颈,研发高效率的Transformer训练/推理计算模式 3.负责基于CUDA/Triton,设计实现高性能GPU算子 4.负责进行多节点分布式训练及调优 5.上述职责满足任意之一即可【任职条件】 1.具有使用深度学习框架(Pytorch, PaddlePaddle, MindSpore等)的