蚂蚁集团-编译器开发工程师（全同态编译器方向）-北京/上海

岗位职责

1. 全同态编译器开发：设计并实现高效的编译器框架，支持全同态加密自动编译与优化。开发编译器前端、中间表示（IR）及后端，确保编译器能够高效处理复杂的数学运算和加密逻辑。优化编译器性能，提升生成代码的执行效率和资源利用率。
2. 算子DSL设计与实现：设计并实现领域特定语言（DSL），用于描述全同态加密数学表达式和计算流程。确保DSL具备良好的可扩展性，支持多种硬件平台和算法需求。提供清晰的DSL接口文档和开发者工具链，降低用户使用门槛。
3. 算子优化：针对全同态加密算子进行深度优化，包括但不限于内存访问模式、并行化策略和硬件加速。分析现有算子的性能瓶颈，提出并实施优化方案，显著提升计算效率。探索新型算法和硬件架构（如GPU、TPU、FPGA等）在算子优化中的应用。
4. 跨团队协作：与算法团队、硬件团队及产品团队紧密合作，确保编译器和算子优化方案满足实际业务需求。参与技术讨论和代码评审，推动团队技术水平的整体提升。
5. 技术研究与创新：跟踪全同态加密领域的最新研究成果，探索其在编译器优化中的潜在应用。发表相关技术论文或专利，提升公司在行业内的技术影响力。

任职要求

1. 、计算机科学、软件工程、数学或相关领域的硕士及以上学历。

2. 、3年以上编译器开发或相关领域的工作经验，有全同态加密或AI编译器开发经验者优先。熟悉LLVM、MLIR或其他主流编译器框架，具备实际项目经验。

3. 精通C/C++、Python等编程语言，熟悉编译原理和编译器优化技术。对全同态加密算法或AI模型的计算特点有深刻理解，能够针对性地优化编译器和算子。熟悉高性能计算（HPC）技术，了解GPU、TPU等硬件加速平台的开发和优化方法。具备DSL设计经验，能够快速实现高效且易用的领域特定语言。

4. 出色的问题分析和解决能力，能够在复杂场景下快速定位问题并提出解决方案。良好的团队合作精神，善于沟通和分享技术经验。对新技术充满热情，具备较强的学习能力和创新意识。 加分项：

5. 在顶级学术会议发表过相关论文。拥有全同态加密或AI编译器相关的开源项目贡献经验。

6. 熟悉PyTorch、TensorFlow、JAX等主流深度学习框架的内部实现。

岗位标签

NEW

特色标签

C、C++、FHE、FPGA加速、GPU加速、HPC优化、LLVM基础设施、LLVM编译器框架、MLIR编译器框架、Python、TPU加速、中间表示、全同态加密技术、全同态加密编译器、加密算子DSL、区块链、同态加密编译器、多级中间表示、安全、密码、支付、算子领域特定语言、编译器中间表示、编译器基础设施、编译器开发经验、编译器构造原理、编译器系统、编译技术、金融、隐私计算、高性能计算