Description

Description

We're looking for an engineer to lead the design, development, and scaling of our machine learning infrastructure. This role is ideal for someone who thrives at the intersection of systems engineering and applied machine learning. You'll be responsible for building robust, scalable, and maintainable infrastructure to support the full ML lifecycle - from data ingestion and feature computation to training, deployment, and monitoring in production. You'll play a critical role in: - Designing and maintaining high-throughput, low-latency pipelines for real-time and batch inference. - Automating the model training and evaluation workflows with reproducibility and traceability in mind. - Defining infrastructure standards and best practices for ML experimentation, CI/CD, and observability. - Collaborating with ML researchers and engineers to improve productivity through tooling and platform enhancements. You thrive in fast-paced, dynamic environments and are comfortable navigating ambiguity to deliver meaningful, incremental impact. You bring strong problem-solving skills, operate with a high degree of autonomy and have a track record of executing effectively. With a commitment to continuous learning and attention to detail, you actively seek opportunities to innovate and share knowledge. You follow engineering best practices - including unit testing, CI/CD, documentation, monitoring, and alerting - to ensure reliable, maintainable solutions.

Minimum Qualifications

• 7 years of development experience and Bachelors or Masters degree in Computer Science or 5 years development experience and PhD in Computer science or related field, with at least 3 years focused on large-scale machine learning infrastructure
• Proficient in Python with solid knowledge of software design principles.
• Expertise in designing and implementing distributed systems or data pipelines (e.g., Spark, Flink, Kafka, Airflow) and knowledge of SQL to analyze data and derive insights.
• Experience with ML lifecycle tools (e.g., MLflow, Metaflow, Kubeflow, SageMaker, Vertex AI).
• Hands-on experience with container orchestration and cloud-native services (e.g., Kubernetes, Docker, AWS/GCP/Azure).
• Leadership experience, including being a technical lead for complex, cross functional development projects demonstrating good technical judgement and prioritization skills. Strong communication skills and a proactive, ownership-driven mindset.

Preferred Qualifications

• Prior experience architecting ML platforms or Feature Stores in a fast-paced production environment.
• Experience with real-time model serving and streaming pipelines (e.g., Kafka, Flink, Ray Serve, Triton).
• Experience optimizing GPU and CPU resource allocation for training and inference workloads.
• Experience with any ML authoring framework (PyTorch, TensorFlow, JAX, etc.), particularly on-device ML frameworks such as CoreML, TFLite or ExecuTorch.

Pay & Benefits

At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $181,100 and $318,400, and your base pay will depend on your skills, qualifications, experience, and location.

Apple employees also have the opportunity to become an Apple shareholder through participation in Apple's discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple's Employee Stock Purchase Plan. You'll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses - including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. Learn more about Apple Benefits.

Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.

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 .