Software Engineer, ML platform and Infrastructure

Apple Inc

Quick summary

Work type
On-site
Location
Austin, TX
Posted
63 days ago

Market check

Salary context

How this pay compares to similar roles

Similar $180k
$135k most similar roles pay here $235k

This listing doesn't post a salary. Most similar roles pay $145,000–$214,850.

Based on 240 similar postings.

Employer

About Apple Inc

Apple Inc. is a multinational technology company known for designing and manufacturing consumer electronics, software, and online services, including the iPhone, Mac, iPad, and App Store. Industry: Consumer Electronics & Software

Apple Inc currently has 1723 open roles on FindRole.

Listed pay typically runs $162,500–$272,100 across 1398 roles with salary data.

Most-posted roles

View all roles at Apple Inc

At a glance

TL;DR · Software Engineer, ML platform and Infrastructure

As a Software Engineer on Apple's Applied Machine Learning team, you will play a pivotal role in designing and building robust, high-performance platforms that support large-scale distributed systems for machine learning initiatives. Your day-to-day responsibilities include architecting scalable ML/Data/Inference platforms, optimizing GPU, TPU, and Inferentia utilization, and integrating cutting-edge open-source technologies to enhance reliability and efficiency. You will work with Kubernetes in production environments on AWS or GCP, contribute to complex open-source codebases, and collaborate closely with cross-functional teams to influence technical strategy. Proficiency in Python, Java, or Go, along with expertise in frameworks like LangGraph and LangChain, is essential, as is a solid background in DevOps practices and security principles for building highly available systems. This role demands deep knowledge of distributed data technologies such as Spark, Flink, Iceberg, and Snowflake to address the unique challenges of scaling ML workloads at Apple's enterprise level.

What you'll do

  • Design and build robust, high-performance platforms for large-scale distributed systems.
  • Engineer systems to optimize utilization and performance of GPUs, TPUs, and Inferentia.
  • Lead the evaluation and integration of cutting-edge open-source technologies in ML infrastructure.
  • Architect containerized workloads using Kubernetes in production environments on AWS/GCP.
  • Build highly available and resilient systems with expertise in operating system internals and networking.
  • Identify and resolve critical system bottlenecks and failures in high-stakes environments.

What we're looking for

  • 5+ years of software development experience focusing on backend systems and APIs.
  • Proficient in Python, Java, or Go, with expertise in building production-grade automation and tooling.
  • Deep hands-on experience with LLM infrastructure requirements and optimizing GPU/TPU utilization.
  • Expertise in designing and building Agents and MCP servers using frameworks like LangGraph and LangChain.
  • Strong background in software engineering for large-scale distributed systems and DevOps practices.
  • Proven ability to architect and engineer containerized workloads using Kubernetes in production environments.
  • Solid understanding of operating system internals, networking protocols, and security principles.

More like this

Similar roles

ML Platform Engineer

Apple Inc

Sunnyvale, CA 56 days ago $147,400$272,100
Python PyTorch TensorFlow JAX Docker Kubernetes CI/CD AWS GCP Azure Spark CoreML Metal CUDA OpenCL Swift C++ Terraform Prometheus