SWE - GPU ML工程师,GGD

15个月前全职
Apple

Apple

location 伦敦
unsaved
摘要 发布日期:2024年1月19日 职位编号:200536347 苹果的计算框架团队为iOS、macOS和Apple TV内外的开发人员提供一套高性能的数据并行算法。我们目前的工作重点是线性代数、图像处理、机器学习和光线追踪等关键领域,以及其他对苹果至关重要的项目。我们一直在寻找非常敬业的个人来壮大我们的优秀团队。 关键资格 关键资格 • 具备编程和解决问题的能力 • 对机器学习基础知识有良好的理解 • 具备系统级编程和计算机架构的经验 • 具备数学背景,包括线性代数和数值方法。 • 具备高性能并行编程经验,有GPU编程经验或LLVM/MLIR编译器基础的经验者优先。 • 具备向机器学习库(如Tensorflow、PyTorch或JAX)添加计算图支持、运行时或设备后端支持的经验者优先。 描述 描述 我们的团队正在寻找对提供强大的计算解决方案以加速苹果硅上的机器学习网络充满热情的非凡的机器学习和GPU编程工程师。工作内容包括在Metal Performance Shaders中定义和实现API,研究新算法,在GPU上为高级机器学习库添加支持,在机器学习计算图中添加优化,进行深入分析、编译器和内核级优化,以确保在硬件系列上实现最佳性能。该职位有机会影响下一代GPU架构中的计算和编程模型的设计。 如果您对此感兴趣,我们很乐意听到您的声音! 教育与经验 教育与经验 技术学士/硕士学位及等同经验。 附加要求 附加要求 • 良好的沟通和协作能力 • 在规定时间内构建高性能、生产质量软件的丰富经验。 • 我们是一家机会均等的雇主,重视公司的多样性。我们不以种族、宗教、肤色、国籍、性别、性取向、年龄、婚姻状况、退伍军人状况或残疾状况为由进行歧视。苹果是一家致力于包容和多样性的机会均等雇主。我们还采取积极措施,向所有申请人提供就业和晋升机会,包括少数民族、女性、受保护的退伍军人和残疾人士。苹果不会歧视或报复对薪酬或其他申请人薪酬进行询问、披露或讨论的申请人。