Job description

In this role, you will push the boundaries of what’s possible on device under tight compute and memory constraints. You will tackle problems such as 3D scene reconstruction and novel view synthesis, generative image and video editing, camera pose estimation, person segmentation, monocular depth estimation, shallow depth of field rendering, and subject tracking. The ideal candidate is focused on product and experience, takes ownership of ambiguous problems, and has a strong foundation in machine learning and graphics. You survey the state of the art before committing to a direction, build working prototypes quickly, and reach out to partner teams to close technical gaps. BS in Computer Science, Electrical Engineering, or related field. Strong programming skills in Python, with working knowledge of C or C++. Hands-on experience with modern deep learning frameworks such as PyTorch or JAX. Solid foundation in machine learning, computer vision, 3D graphics, or related domains. Demonstrated interest in keeping up with state-of-the-art research. Strong interpersonal and communication skills. MS in Computer Science, Machine Learning, Computer Graphics, or related field. Experience with 3D reconstruction, neural rendering such as Gaussian splatting, or generative models for inpainting and outpainting. Experience with camera pose estimation, SLAM, or AR frameworks such as ARKit. Exposure to GPU programming with Metal, CUDA, or similar. Experience with Swift or Objective-C on Apple platforms. Fluency with AI-assisted and agentic coding tools. Track record of successful industry internships.

Requirements

Must-haves

BS in Computer Science, Electrical Engineering, or related field.
Strong programming skills in Python, with working knowledge of C or C++.
Hands-on experience with modern deep learning frameworks such as PyTorch or JAX.
Solid foundation in machine learning, computer vision, 3D graphics, or related domains.
Demonstrated interest in keeping up with state-of-the-art research.
Strong interpersonal and communication skills.
MS in Computer Science, Machine Learning, Computer Graphics, or related field.
Experience with 3D reconstruction, neural rendering such as Gaussian splatting, or generative models for inpainting and outpainting.
Experience with camera pose estimation, SLAM, or AR frameworks such as ARKit.
Exposure to GPU programming with Metal, CUDA, or similar.
Experience with Swift or Objective-C on Apple platforms.
Fluency with AI-assisted and agentic coding tools.
Track record of successful industry internships.

Nice-to-haves

None flagged.

Application prep

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

Keywords to mirror

machine learningpythonswiftai

Platform: apple
Effort: medium

Benefits

Benefits detected in the listing.

Vision