Machine Learning / Computer Vision Engineer - Generative AI

Apple's Synthetic Data Group (SDG) is seeking a skilled Machine Learning Engineer with hands-on experience in Generative AI to join our team. In this role, you will be responsible for designing, training, and deploying state-of-the-art generative models to solve problems across various domains (text, image and video ). You will work closely with cross-functional teams, including technical artists, computer graphics engineers, ML engineers and product managers, to develop cutting-edge AI solutions that push the boundaries of creativity and automation. The ideal candidate will have a proven track record of delivering successful computer vision applications, with experience in developing and deploying deep learning models in production environments.

Description Design, train, and fine-tune Generative AI models (e.g., GANs, VAEs, Diffusion Models, Transformers like GPT, Stable Diffusion). Develop and optimize large-scale deep learning models for text, image and video generation. Implement efficient training pipelines using distributed computing and cloud-based infrastructure (e.g., AWS, Google Cloud Platform). Research and experiment with foundation models, multimodal AI, and reinforcement learning to enhance generative capabilities. Collaborate with synthetic data engineers to preprocess datasets, create synthetic data, and optimize data pipelines. Deploy generative AI models into production using ML frameworks (TensorFlow, PyTorch) and MLOps best practices. Optimize model inference for latency and efficiency, leveraging quantization, pruning, and edge deployment strategies. Stay updated with cutting-edge AI research, contribute to patents/publications, and participate in open-source projects. Work closely with product teams to integrate generative AI into real-world applications (e.g., content creation, automation, digital assistants).

Minimum Qualifications

• BS and a minimum of 3 years relevant industry experience.
• Experience in Computer Graphics fundamentals (3D Gaussian Splats, path tracing, rendering, ...)
• Familiarity with real time / path tracing renderers (VRay, Arnold, Unreal Engine, ...).

Preferred Qualifications

• Experience with RLHF (Reinforcement Learning with Human Feedback).
• Contributions to open-source ML projects or publications in top AI conferences.
• Experience in automated content generation, synthetic media, or AI-powered creative tools.
• 5+ years of hands-on experience in Machine Learning & Deep Learning.
• Strong expertise in Generative AI models (Diffusion Models, GANs, VAEs, Transformers).
• Experience using Synthetic Data for training computer vision models
• Proficiency in Python and ML frameworks like TensorFlow, PyTorch, JAX.
• Experience with NLP, Computer Vision, or Multimodal AI.
• Familiarity with LLMs (GPT, LLaMA, Mistral, etc.) and fine-tuning techniques like LoRA, PEFT.
• Experience in model deployment using Docker, TensorRT, ONNX.
• Strong knowledge of cloud computing (AWS, Google Cloud Platform) and distributed model training.
• Experience with MLOps, model versioning, and monitoring (MLflow, Weights & Biases).
• Strong problem-solving skills and ability to work in a fast-paced, product-focussed environment.

Pay & Benefits At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $143,100 and $264,200, and your base pay will depend on your skills, qualifications, experience, and location.

Apple employees also have the opportunity to become an Apple shareholder through participation in Apple's discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple's Employee Stock Purchase Plan. You'll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses - including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. Learn more about Apple Benefits.

Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.

Apple is an equal opportunity employer that is committed to inclusion and diversity. We take affirmative action to ensure equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant.