岗位职责
- 全同态编译器开发:设计并实现高效的编译器框架,支持全同态加密自动编译与优化。开发编译器前端、中间表示(IR)及后端,确保编译器能够高效处理复杂的数学运算和加密逻辑。优化编译器性能,提升生成代码的执行效率和资源利用率。
- 算子DSL设计与实现:设计并实现领域特定语言(DSL),用于描述全同态加密数学表达式和计算流程。确保DSL具备良好的可扩展性,支持多种硬件平台和算法需求。提供清晰的DSL接口文档和开发者工具链,降低用户使用门槛。
- 算子优化:针对全同态加密算子进行深度优化,包括但不限于内存访问模式、并行化策略和硬件加速。分析现有算子的性能瓶颈,提出并实施优化方案,显著提升计算效率。探索新型算法和硬件架构(如GPU、TPU、FPGA等)在算子优化中的应用。
- 跨团队协作:与算法团队、硬件团队及产品团队紧密合作,确保编译器和算子优化方案满足实际业务需求。参与技术讨论和代码评审,推动团队技术水平的整体提升。
- 技术研究与创新:跟踪全同态加密领域的最新研究成果,探索其在编译器优化中的潜在应用。发表相关技术论文或专利,提升公司在行业内的技术影响力。
职位要求
- 、计算机科学、软件工程、数学或相关领域的硕士及以上学历。
- 、3年以上编译器开发或相关领域的工作经验,有全同态加密或AI编译器开发经验者优先。熟悉LLVM、MLIR或其他主流编译器框架,具备实际项目经验。
- 精通C/C++、Python等编程语言,熟悉编译原理和编译器优化技术。对全同态加密算法或AI模型的计算特点有深刻理解,能够针对性地优化编译器和算子。熟悉高性能计算(HPC)技术,了解GPU、TPU等硬件加速平台的开发和优化方法。具备DSL设计经验,能够快速实现高效且易用的领域特定语言。
- 出色的问题分析和解决能力,能够在复杂场景下快速定位问题并提出解决方案。良好的团队合作精神,善于沟通和分享技术经验。对新技术充满热情,具备较强的学习能力和创新意识。 加分项:
- 在顶级学术会议发表过相关论文。拥有全同态加密或AI编译器相关的开源项目贡献经验。
- 熟悉PyTorch、TensorFlow、JAX等主流深度学习框架的内部实现。