Machine Learning System Software Engineer

Job description

At Apple, we're on the cutting edge of delivering transformative experiences through Artificial Intelligence. If you're passionate about pushing the boundaries of AI and hardware optimization, we want you to join our team! As a Machine Learning System Software Engineer on the Apple Neural Engine (ANE) team, you'll work to bring high-performance, low-power AI solutions to life on iconic Apple products like the Vision Pro, iPhone, iPad, Mac, and more. Minimum Qualifications: BS and a minimum of 10 years relevant industry experience Experience defining interfaces that are used by other teams or external developers, with attention to lifecycle, error handling, and forward compatibility Deep proficiency in C and C++ in large, production system software Understanding of runtime systems: process/thread models, memory management, IPC/RPC, and resource lifecycle Understanding of software-hardware interfaces: registers, DMA, command queues, or similar accelerator interaction patterns Experience shipping production system software Preferred Qualifications: Experience building or extending ML runtimes, inference engines, or accelerator driver stacks (e.g., TensorRT, ONNX Runtime, XLA, Metal, Vulkan compute, or similar) Familiarity with Swift/Rust or another memory-safe systems language Familiarity with ML model compilation pipelines and how runtime APIs interact with compiler outputs (graph IR, compiled binaries) Experience with multi-client runtime scenarios: arbitrating hardware access, managing priority/QoS, and handling client lifecycle (ex: connect, disconnect, crash recovery) Knowledge of neural network inference: operator execution, tensor memory layout, pipelining, and batching strategies Strong communication skills and experience working across team boundaries (framework teams, compiler teams, hardware teams)