岗位职责:
工作领域为从事AI推理框架的开发,具体包括以下内容。
1. 针对不同AI框架的模型,把其转换到特定的IR上。
2. 在AI计算图上开发各种优化算法,包括机器无关优化和机器相关优化。
3. 根据不同target的ISA描述,在性能评估的结果的基础上,进行代码生成,把计算图中的不同部分转化为不同target上的可执行代码。
4. 针对特定框架,开发高性能AI算子。
任职要求:
1. 要求有2年以上AI软件栈方面的开发经验。
2. 对存储优化、并行优化等优化原理有一定程度的掌握。
3. 熟悉AI推理的优化过程,参与过基于TVM、MLIR或者其他自研框架下推理引擎的开发,有机器相关优化经验者优先。
4. 掌握用CUDA/OpenCL编写算子的技术,有具体开发经验者优先。