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

Apple, Inc.; Austin (TX); On-site; USD 140,000 - 200,000; Full time

Job summary

Apple, Inc. in Austin, TX is seeking a Software Engineer (AML) for AI & Data Platforms to build scalable ML systems that power Apple products and services. You will design ML infrastructure, feature engineering pipelines, and model serving components for production use.

The role emphasizes experimentation with transformers, NLP, CV, and data visualization tools, with strong Python and ML framework experience. A BS in CS and 3+ years in ML engineering are required.

Qualifications

• BS in Computer Science or related field (or equivalent).
• 3+ years of ML engineering experience in feature engineering, model training, serving, monitoring, and refresh management.
• Experience building ML systems at scale in production or high-impact research environments.
• Strong interest in CV/NLP and LLM/Generative AI systems; familiar with PyTorch or TensorFlow.

Responsibilities

• Design and build ML models and scalable ML infrastructure for production.
• Develop feature engineering, data augmentation, and model evaluation pipelines.
• Experiment with transformer models (e.g., BERT, GPT) and fine-tuning frameworks.
• Collaborate with data and product teams to deploy AI solutions across apps and services.

Skills

ML engineering

Feature engineering

Model training

Model serving

Model monitoring

MLOps

Education

Bachelor's degree in Computer Science or related field

Tools

PyTorch

TensorFlow

Python

SQL

Looker

Tableau

Superset

Streamlit

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