天数智芯 - 做算力引擎,创世界一流

加入我们

首页 加入我们 软件工程师 – 计算框架
软件工程师 – 计算框架
上海/西安/南京

岗位职责

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等推理引擎,了解原理。

申请工作
分享此职位
软件工程师 – 驱动
2020.12.31
上海/西安/南京
软件工程师 – 编译器
2020.12.31
上海/西安/南京
Core Design Engineer
2021.01.06
上海

加入我们

我们的核心价值观

自强、创新、求真、匠心、包容。