在这个职位上,您将成为MTIA(元训练与推理加速器)软件团队的一员,并成为领先行业的PyTorch AI框架组织的一部分。MTIA软件团队一直在开发一种全面的AI编译器策略,为专用硬件架构上的生产环境提供高度灵活的平台,用于训练和服务新的DL/ML模型架构,并结合自动调优的高性能。编译器堆栈、DL图优化和特定硬件的内核编写直接影响Meta上AI训练和推理平台的性能和部署速度。您将在PyTorch框架组件、AI编译器和运行时、高性能内核和工具等核心领域之一上工作,以加速当前和下一代MTIA AI硬件平台上的机器学习工作负载。您将与AI研究人员紧密合作,分析深度学习模型并在MTIA硬件上高效降低它们。您还将与硬件设计团队合作,为高性能开发编译器优化。您将应用软件开发最佳实践来设计功能、优化和性能调优技术。您将获得在开发机器学习编译器框架方面的宝贵经验,并帮助推动AI领域特定问题的下一代硬件软件协同设计。
所需技能:
软件工程师,系统ML-框架/编译器/内核职责:
1. 开发与以下核心关注领域之一相关的软件堆栈:AI框架、编译器堆栈、高性能内核开发和加速到下一代硬件架构。
2. 为开发业界领先的PyTorch AI框架核心编译器,支持新的最先进的推理和训练AI硬件加速器,并优化其性能做出贡献。
3. 分析深度学习网络,开发和实施编译器优化算法。
4. 与AI研究科学家合作,加速下一代深度学习模型,如推荐系统、生成式AI、计算机视觉、自然语言处理等。
5. 对深度学习框架和软件组件进行性能调优和优化。
最低资格要求:
最低资格要求:
6. 熟练的C/C++编程技能
7. 在AI框架开发或在硬件架构上加速深度学习模型方面的经验。
8. 计算机科学、计算机工程、相关技术领域的学士学位或同等实践经验。
首选资格:
首选资格:
9. 计算机科学、计算机工程、相关技术领域的学士学位,并且在AI框架开发或在硬件架构上加速深度学习模型方面有7年以上的经验;或者计算机科学、计算机工程、相关技术领域的硕士学位,并且在AI框架开发或在硬件架构上加速深度学习模型方面有4年以上的经验;或者计算机科学、计算机工程或相关技术领域的博士学位,并且在AI框架开发或在硬件架构上加速深度学习模型方面有3年以上的经验。
10. 了解GPU、CPU或AI硬件加速器架构。
11. 有使用PyTorch、Caffe2、TensorFlow、ONNX、TensorRT等框架的经验。
12. 或者AI高性能内核:有CUDA编程、OpenMP/OpenCL编程或AI硬件加速器内核编程的经验。在AI硬件上加速库的经验,类似于cuBLAS、cuDNN、CUTLASS、HIP、ROCm等。
13. 或者AI编译器:有编译器优化的经验,如循环优化、矢量化、并行化、特定硬件的优化,如SIMD。有MLIR、LLVM、IREE、XLA、TVM、Halide等经验者优先。
14. 或者AI框架:有开发训练和推理框架组件的经验。有系统性能优化的经验,例如运行时延迟分析、内存带宽、I/O访问、计算利用率分析以及相关工具开发。
公共薪酬:
每年146,994美元至208,000美元+奖金+股权+福利
行业:互联网
平等机会:
Meta自豪地成为一家平等就业机会和积极行动的雇主。我们不以种族、宗教、肤色、国籍、性别(包括怀孕、分娩或相关医疗状况)、性取向、性别认同、性别表达、跨性别身份、性别刻板印象、年龄、作为受保护退伍军人的身份、作为残疾人的个人身份或其他适用的法律保护特征来歧视。我们还会考虑符合适用联邦、州和地方法律的资格要求的申请人。Meta在某些地方参与E-Verify计划,根据法律要求。请注意,Meta可能在就业申请中利用人工智能和机器学习技术。
Meta致力于为我们招聘过程中有残疾的候选人提供合理的住宿。如果您因残疾需要任何帮助或住宿,请通过accommodations-ext@fb.com与我们联系。
关于公司:Meta