AI芯片工具链开发

1. 负责开发面向自研芯片的AI模型编译工具链，支持CNN、Transformer等主流网络的高效部署。 2. 设计并实现模型优化、图编译、算子融合、内存优化等关键技术，提升芯片上的推理性能。 3. 开发自动化工具链，简化模型从训练框架（如PyTorch/TensorFlow）到自研芯片的部署流程。 4. 与芯片团队紧密合作，针对硬件特性（如指令集、内存架构）进行深度优化。 5. 负责工具链的API设计、文档建设和开发者体验优化，推动工具链在客户侧的落地。

必需条件： - 计算机、电子工程、数学等相关专业硕士及以上学历，1年以上AI工具链开发经验。 - 精通Python和C++，具备高性能代码开发和调试能力。 - 熟悉AI模型部署全流程，包括模型量化、编译优化等。 - 深入理解CNN/Transformer等网络结构及其计算特性。 - 熟悉至少一种主流深度学习框架（PyTorch/TensorFlow）。 加分项： - 有TVM开发经验，熟悉TVM Stack（Relax/TIR/Ansor）。 - 有AI算子开发经验，熟悉CUDA/OpenCL或芯片特定指令集优化。 - 参与过完整AI工具链开发项目，产出过成熟部署工具。 - 了解自研芯片架构（如NPU、DSA），能针对硬件特性优化工具链。 #### 我们提供： - 参与前沿AI芯片研发的机会，直接影响工具链的技术方向。 - 技术挑战与成长空间：解决高并发、低延迟、异构计算等难题。 - 有竞争力的薪资+期权/股票+完善的福利体系。 - 灵活的工作地点：北京、上海、深圳、西安和成都都可选。