Software Engineer, Machine Learning & AI

Apple

  • Shanghai
  • Permanent
  • Full-time
  • 23 days ago
Are you passionate about Machine Learning and AI, and eager to apply your expertise to solve real-world problems at Apple's incredible scale? Do you want to see your work translated into direct impact on iPhone, AirPods, Watch, Mac, and more? Are you a motivated self-starter eager to lead projects while helping build the culture of a rapidly growingteam? Come join us in applying AI and ML techniques across Apple's hardware portfolio, helping bring amazing, groundbreaking products and innovations to life. We are the Product Integrity AI/ML team, and we develop and deliver software that supports the creation of Apple's unparalleled hardware product line. Our software is used by both Apple engineers and third-party accessory manufacturers to develop and debug their systems, with high visibility throughout the company. If you're passionate about writing clean code with a focus on delivering maximum customer impact, we can't wait to hear from you!DescriptionAs a Software Engineer on our team, you will play a pivotal role in shaping and delivering high-impact AI/ML solutions that redefine how Apple designs, manufactures, and tests hardware products. The software you build will power data-driven decisions, uncover new insights, and enable breakthrough capabilities across Hardware Engineering and beyond. We operate in a fast-paced, collaborative environment, working closely with cross-functional teams to quickly iterate, validate, and deploy practical AI/ML solutions at scale. Key Responsibilities: -Design, implement, and maintain AI/ML software solutions with clean, scalable, and testable code -Collaborate with senior engineers and cross-functional partners to understand requirements and translate them into concrete technical tasks -Develop and refine ML workflows, including training, evaluation, and inference pipelines for both traditional models and LLM-based solutions -Contribute to agentic system development and RAG-based architectures that enable intelligent automation and reasoning -Stay up to date with the evolving AI/ML landscape and help improve our tools, infrastructure, and practices based on new developments -Participate in code reviews, design discussions, and documentation to ensure high engineering quality and team knowledge sharing -Support testing, deployment, and monitoring of models and services in production environmentsMinimum Qualifications
  • 3+ years of professional software engineering experience, with a focus on AI/ML development
  • Proficiency in Python or another object-oriented programming language
  • Hands-on experience with training and deploying ML models or using LLMs in applications
  • Strong problem-solving skills and an ability to learn quickly in a fast-paced environment
  • Bachelor's degree in Computer Science, Engineering, or a related field, or equivalent industry experience
  • Strong communication skills and a collaborative mindset
Preferred Qualifications
  • Exposure to agentic workflows, RAG pipelines, or LLM APIs (e.g., OpenAI, HuggingFace, LangChain)
  • Experience with ML infrastructure tools (e.g., PyTorch, TensorFlow, Ray, Airflow)
  • Familiarity with containerization (e.g., Docker), CI/CD, and cloud-based development
  • Passion for continuous learning, knowledge-sharing, and improving how teams build AI/ML systems

Apple