Job description

This role requires an independent, self-motivated, and creative engineer with deep expertise in machine learning, coupled with a strong understanding of 3D computer vision and computer graphics. Your primary focus will be applying cutting-edge machine learning techniques to complex 3D visual challenges to create customer impact across current and future Apple products. Bachelor's degree in Computer Science, Electrical/Computer Engineering, or a related field. 10+ years of experience developing and shipping machine learning models, with a focus on computer vision or generative AI. Expert-level proficiency in Python and PyTorch or equivalent ML frameworks. Strong experience with diffusion models for image synthesis, video generation, 3D generation, or related domains. Solid understanding of 3D computer vision including depth estimation, multi-view geometry, 3D reconstruction, novel view synthesis, or neural scene representations. Working knowledge of 3D computer graphics concepts such as rendering, camera models, scene representations, and coordinate systems. Working knowledge of C++ for performance-critical components. Ability to leverage AI coding assistants and LLM-based tools to accelerate development, prototyping, and problem-solving workflows. MS or PhD in Computer Science, Machine Learning, Computer Vision, or a related field. Publications in top-tier ML or vision venues. Experience with video diffusion models, 3D Gaussian Splatting, neural radiance fields, or differentiable rendering. Experience with distributed training at scale. Familiarity with on-device ML deployment. Track record of shipping ML-powered features in consumer products. Good written and oral interpersonal skills.

Requirements

Must-haves

Bachelor's degree in Computer Science, Electrical/Computer Engineering, or a related field.
10+ years of experience developing and shipping machine learning models, with a focus on computer vision or generative AI.
Expert-level proficiency in Python and PyTorch or equivalent ML frameworks.
Strong experience with diffusion models for image synthesis, video generation, 3D generation, or related domains.
Solid understanding of 3D computer vision including depth estimation, multi-view geometry, 3D reconstruction, novel view synthesis, or neural scene representations.
Working knowledge of 3D computer graphics concepts such as rendering, camera models, scene representations, and coordinate systems.
Working knowledge of C++ for performance-critical components.
Ability to leverage AI coding assistants and LLM-based tools to accelerate development, prototyping, and problem-solving workflows.
MS or PhD in Computer Science, Machine Learning, Computer Vision, or a related field.
Publications in top-tier ML or vision venues.
Experience with video diffusion models, 3D Gaussian Splatting, neural radiance fields, or differentiable rendering.
Experience with distributed training at scale.
Familiarity with on-device ML deployment.
Track record of shipping ML-powered features in consumer products.
Good written and oral interpersonal skills.

Nice-to-haves

None flagged.

Application prep

Use the role’s language and application details while tailoring your materials.

Keywords to mirror

machine learningaipythonmlllm

Platform: apple
Effort: medium

Benefits

Benefits detected in the listing.

Vision