AI ML Platform Jobs in Austin, TX

The Applied Machine Learning team has been at the forefront of accelerating digital transformation through machine learning across Apple's enterprise ecosystem. Our ML Platforms, Solutions, and Services deliver a comprehensive suite of capabilities that drive efficiency, agility, and innovation at Apple scale-serving business-critical needs across the enterprise.

We are looking for talented Software Engineers who are passionate about distributed systems and large-scale infrastructure to build and operate world-class ML platforms and products across cloud environments.

Description

Join Apple's Applied Machine Learning Team as a Machine Learning Platform Engineer and play a central role in designing and building the systems that power our Data, Machine Learning, and Generative AI initiatives. You will architect and engineer robust, high-performance, massively scalable platforms that serve as the foundation for groundbreaking ML workloads across the enterprise.

In this role, you will apply software engineering depth to solve the hardest challenges in large-scale distributed systems-designing for reliability, performance, and efficiency from the ground up. You will own the technical direction of ML/Data/Inference platform capabilities, leading the evaluation and integration of cutting-edge open-source technologies and building innovative internal solutions that raise the bar for scalability and resilience across our ML ecosystem. You'll collaborate closely with cross-functional engineering and business teams, influencing technical strategy and contributing meaningfully to the broader platform roadmap.","responsibilities":"Highly proficient in Python, Java, or Go, with a strong track record of building production-grade automation, tooling, and system-level software.

Deep understanding of LLM infrastructure requirements-including GPUs, TPUs, and Inferentia-with hands-on experience engineering systems that optimize their utilization and performance.

Experience designing and building Agents and MCP servers, with hands-on expertise in frameworks such as LangGraph and LangChain.

Solid background in software engineering for complex, large-scale distributed systems, with strong familiarity with DevOps and reliability engineering practices.

Expert-level proficiency with AWS/GCP and deep, hands-on experience architecting and engineering containerized workloads using Kubernetes in production environments.

Proven ability to read, understand, and make meaningful contributions to complex open-source codebases in the ML infrastructure space.

Strong command of operating system internals, networking protocols, and security principles, applied to building highly available and resilient systems.

Exceptional analytical and problem-solving skills, with a demonstrated ability to identify and resolve critical system bottlenecks and failures in high-stakes environments.

Preferred Qualifications

Experience engineering scalable solutions for data processing and model training/fine-tuning workflows.

Hands-on experience building with distributed data technologies for ML training such as Spark, Flink, Iceberg, or Snowflake, with a deep understanding of their architectural trade-offs at scale.

Minimum Qualifications

5+ years of experience in software development, with a strong focus on backend systems and APIs.

2+ years of experience working with LLMs, Agent Frameworks

5+ years of experience with cloud platforms such as AWS,or GCP

Pay & Benefits

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

"},{"displayOrder":3,"type":"ACCESSIBILITY_RULES","content":"

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

"},{"displayOrder":4,"type":"OTHER_POSTING_SUPPLEMENTS_NEW","content":"

Apple accepts applications to this posting on an ongoing basis.

"}]}},{"postLocationId":"postLocation-SFMETRO","localizations":{"en_US":[{"displayOrder":1,"type":"POSTING_FOOTER_RULES_NEW","content":"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 $212,000 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.