hero

COMPANIES YOU'LL LOVE TO WORK FOR

companies
Jobs

Machine Learning Engineer - AI Automation & Deep Learning

Lex

Lex

Software Engineering, Data Science
Shanghai, China
Posted on Apr 1, 2026
Summary

Join the HWTE Computer Vision and Machine Learning (CVML) team and play a key role in shaping the future of Apple's products! Our team develops cutting-edge solutions for the inspection, metrology, and calibration of displays, cameras, and sensors across iPhone, iPad, Mac, Vision Pro, and upcoming products. Be among the first to witness the creation of Apple's newest products!

We are looking for a versatile engineer who can bridge AI automation with domain expertise in computer vision, image processing, sensor algorithms, or simulation pipelines. In this role, you will build intelligent AI-powered tools and interfaces that make our world-class algorithm and simulation pipelines accessible to every engineer in the organization, automate complex engineering workflows to dramatically improve team efficiency, and contribute to algorithm development for display, camera, motion sensor, or optical testing. Your work will directly multiply the productivity of the engineering team and transform how we operate across every stage of the product lifecycle.

If you are passionate about applying modern AI tools — such as large language models, AI agents, and generative AI — to solve real-world engineering challenges, we want to hear from you!

Description

You will design, develop, and deploy AI-powered tools that democratize access to complex algorithm and simulation pipelines, automate multi-step workflows, and capture expert domain knowledge into durable, reusable systems. You will also contribute hands-on to computer vision, image processing, deep learning, and sensor algorithm development for display, camera, or motion sensor testing.

Responsibilities

  • AI-Powered Interfaces: Build intelligent, LLM-based interfaces that allow engineers to configure and run complex simulation pipelines and algorithm pipelines through natural language — without requiring deep knowledge of the underlying code — and incorporate appropriate validation to ensure correctness of automated outputs.
  • Code Modularization: Modularize and abstract existing algorithm and simulation pipeline codebases to expose clean parameters and make them AI-accessible.
  • Workflow Automation: Automate complex engineering workflows end-to-end — from retrieving multi-modal data (images, factory result tables, field reports) to running analysis, generating visualizations, and triggering appropriate analysis, corrective actions, or further investigations — especially for algorithm and software solutions.
  • Production Monitoring & Triage: Build monitoring and triage tools that automatically ingest production data, detect anomalies or late findings, and kick off the appropriate analysis or corrective actions across all build phases.
  • Code Conversion: Develop AI-assisted tools that convert prototype algorithm code into production-ready, high-performance implementations across multiple languages and deployment targets.
  • Knowledge Capture: Encode expert engineering knowledge into AI systems so that domain expertise becomes durable, transferable, and accessible to the broader team — reducing single-person dependencies.
  • Algorithm Development: Contribute to computer vision, image processing, deep learning, and sensor algorithm development, including the full machine learning lifecycle from data collection and preprocessing to model development, deployment, and monitoring.
  • Cross-Functional Collaboration: Collaborate with domain experts across display, camera, motion sensor, optical sensor, and touch testing to identify workflow bottlenecks and translate domain knowledge into automated solutions.
  • Continuous Improvement: Stay current with AI/ML advancements and evaluate how emerging tools and techniques can be applied to algorithm and simulation pipeline challenges.

Minimum Qualifications

  • 2+ years of related professional experience
  • MS in Computer Science, Electrical Engineering, Imaging Science, Robotics, or a related field
  • Proficiency in MATLAB, Python, or C++ with a solid foundation in software engineering and system design
  • Experience building applications powered by large language models (LLMs), including prompt engineering, retrieval-augmented generation, function calling / tool use, or agentic workflow development
  • Background in at least one of: computer vision, image processing, deep learning, or sensor algorithms
  • Hands-on experience with deep learning frameworks such as PyTorch or TensorFlow
  • Strong written and verbal communication skills, with the ability to collaborate effectively across diverse teams

Preferred Qualifications

  • Hands-on experience with modern AI coding and automation tools (e.g., Claude, Gemini) and familiarity with emerging concepts such as AI agents, skills, agentic workflows, and Model Context Protocol (MCP)
  • Experience with computer vision tasks such as defect detection, image segmentation, image quality assessment or multi-view geometry.
  • Familiarity with display metrology, camera calibration, or inertial measurement units (IMU)
  • Experience working across multiple programming languages and converting or bridging code between them
  • Experience modularizing and refactoring complex codebases to improve usability and maintainability
  • Experience automating multi-step data workflows involving databases, APIs, and visualization tools
  • Familiarity with consumer electronics manufacturing or hardware test engineering

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.