Software Engineer (AML), AI & Data Platforms (AiDP)

Apple Inc.; Austin, Northern (TX, KY); Hybrid; USD 120,000 - 180,000; Full time

Job summary

Apple Inc. in Austin, Texas is seeking a Software Engineer (AML) for AI & Data Platforms (AiDP). You will help build scalable ML systems that power Apple products and services, bridging data, development, and ML workflows.

You will design and deploy ML pipelines, feature engineering, model training, serving, and monitoring, using PyTorch or TensorFlow, Python, and SQL. Strong software engineering practices and collaboration across teams are essential.

Qualifications

• BS in Computer Science or related field.
• 3+ years of ML engineering experience in feature engineering, model training, serving, monitoring, and refresh management.
• Experience at scale in production or high-impact research.
• Passion for computer vision and NLP, especially LLMs and Generative AI systems.
• Knowledge with PyTorch or TensorFlow.

Responsibilities

• Build ML models and ML infrastructure for production systems.
• Develop robust, scalable ML pipelines and model serving components.
• Perform feature engineering, data augmentation, and fine-tuning for domain tasks.
• Collaborate with teams to deploy AI solutions that scale across Apple services.

Skills

Python

SQL

Anomaly detection

Education

BS in Computer Science or related field

Tools

PyTorch

TensorFlow

Superset

Tableau

Looker

Job description

Software Engineer (AML), AI & Data Platforms (AiDP)

Austin, Texas, United States Software and Services

Do you want to help build some of the largest and most consequential enterprise and customer technology systems in the world? Join Apple’s Information Systems and Technology (IS&T) organization. IS&T is the engine behind everything Apple does for customers and for the people who build for them. It’s Apple’s central nervous system. Supporting 2.5 billion active Apple devices, processing billions of secure transactions, and keeping the technology that defines modern life running flawlessly, IS&T makes the impossible feel effortless. Do you love building solutions to handle global complexity and immense scale? Imagine what you could do here. AI & Data Platforms (AiDP) is IS&T's engine for AI-powered innovation. The team brings together data, application development, and machine learning — including generative AI — along with data services and customer success functions, to help IS&T build solutions more efficiently and streamline the adoption and embedding of generative AI across Apple.

Description

As a Machine Learning Engineer, you will work on building intelligent systems to democratize AI across a wide range of solutions within Apple. You will drive the development and deployment of innovative AI models and systems that directly impact the capabilities and performance of Apple’s products and services. You will implement robust, scalable ML infrastructure, including data storage, processing, and model serving components, to support seamless integration of AI/ML models into production environments. You will develop novel feature engineering, data augmentation, prompt engineering and fine-tuning frameworks that achieve optimal performance on specific tasks and domains. You will design and implement automated ML pipelines for data preprocessing, feature engineering, model training, hyper-parameter tuning, and model evaluation, enabling rapid experimentation and iteration. You will also implement advanced model compression and optimization techniques to reduce the resource footprint of language models while preserving their performance. There are massive opportunities for you deliver impactful influences to Apple.

Minimum Qualifications

• BS in Computer Science or related field or equivalent.
• 3+ years of machine learning engineering experience in feature engineering, model training, model serving, model monitoring and model refresh management.
• Experience developing AI/ML systems at scale in production or in high-impact research environments.
• Passionate about computer vision, natural language processing, especially in LLMs and Generative AI systems.
• Knowledge with the common frameworks and tools such as PyTorch or TensorFlow.

Preferred Qualifications

• Experience in Anomaly detection and forecasting & related methodologies
• Experience and proficiency in python & writing efficient SQLs.
• Strong coding and software engineering skills, and familiarity with software engineering principles around testing, code reviews and deployment.
• Proven experience with transformer models such as BERT, GPT etc., and a proven understanding of their underlying principles is a plus
• Data Visualization Tools: Proficient in data visualization, with experience in software such as Superset, Streamlit, Tableau, Business Objects, and Looker

Apple is an equal opportunity employer that is committed to inclusion and diversity. We seek to promote 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

At Apple, we believe accessibility is a fundamental human right. You’ll find that idea reflected in everything here — in our culture, our benefits and our digital tools. By welcoming as many perspectives as possible, we help you build a career where you feel like you belong.

Learn about accessibility in Apple’s workplace

Learn about reasonable accommodations for job applicants

Apple accepts applications to this posting on an ongoing basis.