ML Engineer, AI & Data Platforms (GenAI)

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

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.