Computer Vision Algorithm Engineer

We are a team at Apple developing state-of-the-art generative AI and 3D computer vision technologies for next-generation content creation systems. We are looking for engineers with deep expertise at the intersection of 3D vision, diffusion models, and low-level performance optimization.
In this role, you will collaborate with cross-functional research and product teams to advance generative vision models while building high-performance systems that bring cutting-edge technologies into Apple products.

Description
As part of the team, you will design, implement, and optimize 3D vision algorithms and diffusion-based generative models across Apple platforms, including iOS and macOS.
You will work across the full stack of modern generative AI systems-from model development and training to low-level kernel implementation and performance optimization. The role requires a strong understanding of 3D computer vision, diffusion architectures, generative modeling, and efficient training and inference systems.

Preferred Qualifications
Experience with GPU domain-specific languages such as Triton, Taichi, CuTe, or similar kernel programming frameworks
Experience training large-scale models, such as diffusion models or large language models
Experience deploying 3D vision algorithms or diffusion models on mobile or resource-constrained platforms
Strong interest in systems optimization, software architecture, and performance engineering
Self-motivated and curious about emerging technologies in generative AI
Apple is an equal opportunity employer that is committed to inclusion and diversity, and thus we treat all applicants fairly and equally. Apple is committed to working with and providing reasonable accommodation to applicants with physical and mental disabilities

Minimum Qualifications
PhD/MS in Computer Vision, Machine Learning, or a related field, or equivalent industry experience delivering production ML systems
Research or industry experience in 3D computer vision and generative models
Experience with training feed-forward or generative models
Experience with GPU kernel development and performance optimization (e.g., Metal, CUDA, or similar parallel computing frameworks)

Apple is an equal opportunity employer that is committed to inclusion and diversity, and thus we treat all applicants fairly and equally. Apple is committed to working with and providing reasonable accommodation to applicants with physical and mental disabilities.