Gen AI Research Engineer – Diffusion & Foundation Models

Apple Inc.; Cupertino, Northern (CA, KY); Hybrid; USD 216,000 - 325,000; Full time

Benefits offered by this job

Stock options

Relocation assistance

Education reimbursement

Medical and dental coverage

Job summary

Apple Inc. in Cupertino, California, is seeking a machine learning research engineer to advance diffusion- and autoregressive-based generative models for on-device experiences. You will train and adapt large-scale foundation models and publish results that push the boundaries of AI within Apple’s ecosystem.

The role emphasizes collaboration across teams to ship state-of-the-art tech that enhances user creativity and interaction on Apple devices.

Qualifications

• MS or PhD in Computer Science or related field.
• 2+ years training and fine-tuning modern image/video generation models.
• Experience in academic or industrial research environment.
• Proficient in PyTorch, TensorFlow, and JAX.
• Excellent communication and ability to drive clarity in ambiguous situations.

Skills

PyTorch

TensorFlow

JAX

Communication

On-device optimization

Education

MS/PhD in CS or related field

Job description

Apple Inc. in Cupertino, California, is seeking a machine learning research engineer to advance diffusion- and autoregressive-based generative models for on-device experiences. You will train and adapt large-scale foundation models and publish results that push the boundaries of AI within Apple’s ecosystem.

The role emphasizes collaboration across teams to ship state-of-the-art tech that enhances user creativity and interaction on Apple devices.