岗位职责

1. 基于自研GPGPU适配/优化/维护常用AI计算框架TensorFlow, PyTorch, MXNet等；
2. 实现易扩展、高性能、大规模的AI模型训练系统，发挥自研GPGPU的优越性；
3. 基于自研芯片设计构建高性能推理引擎。

任职资格

1. 具有计算机相关领域背景，熟悉AI领域更好；
2. 熟练使用Python/C/C++编程；
3. 熟悉CUDA编程；
4. 对TensorFlow/PyTorch实现原理熟悉、熟悉底层代码的实现；
5. 对分布式计算/训练框架有开发经验；
6. 熟悉TensorRT/TVM等推理引擎，了解原理。