Machine Learning System Software Engineer

Apple • Sunnyvale, CA

75d

About The Position

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. This is a dynamic opportunity to work in a creative, collaborative environment while developing groundbreaking technologies that will shape the future of computing! We are looking for an engineer with deep expertise in system software technology who is eager to tackle new challenges and responsibilities as the role evolves. As the position progresses, there will be opportunities to demonstrate technical leadership, influence key design decisions, collaborate with and support other engineers, and help guide the direction of Apple's AI-driven capabilities across the ecosystem.

Requirements

• Experience optimizing software for distributed, parallel, or heterogeneous execution environments, with a solid understanding of shared memory, synchronization, and multi-threading techniques
• Expertise in neural network inference on specialized SoCs or GPUs, and knowledge of deep learning frameworks and tools
• Expertise in full stack software design and implementation
• Proven track record in mentoring and coaching engineers, with an interest in taking on increasing responsibilities and contributing to the team's development
• Strong communication skills and ability to collaborate effectively across teams and functions

Responsibilities

• Architect and develop the system runtime software for Apple's proprietary Neural Engine Accelerator, optimizing it for deep learning inference with a focus on performance, scalability, and power efficiency
• Collaborate with cross-functional teams, including hardware and platform architecture teams, to bring new hardware silicon to market and ensure the system software support for next-gen features
• Lead the design and implementation of complex system runtime features, advancing both technical capabilities and strategic alignment across the team and company
• Mentor and guide emerging and mid-level engineers, sharing your expertise to foster their growth in system software technologies and ensuring high standards of code quality, innovation, and efficiency
• Play an instrumental role in defining system software architecture approaches and optimizations, balancing trade-offs between performance, energy efficiency, and hardware constraints
• Identify and drive initiatives that will improve the scalability and general performance of AI workloads on Apple hardware, contributing to the vision and roadmap of the Apple Neural Engine team
• Cultivate a culture of learning and collaboration, encouraging innovation, and supporting engineers in their development, with the potential to expand your influence across the organization

What This Job Offers

• Career Level: Mid Level
• Industry: Computer and Electronic Product Manufacturing
• Number of Employees: 5,001-10,000 employees