AIML - Software Engineer for MLX, MLR

Apple's Machine Learning Research organization is at the forefront of AI innovation, conducting world-class scientific research and building technologies that power future Apple products. The team's work impacts ML solutions across Apple, affecting features used by billions of consumers worldwide.

As a Software Engineer on the MLX team, you'll be part of a dynamic group focused on advancing ML algorithms and techniques on Apple silicon. You'll collaborate with researchers and software engineers to build scalable, distributed training and research pipelines. The role requires expertise in high-performance computing, particularly in GPU programming and ML algorithm implementation.

Key responsibilities include developing high-performance software for novel ML research algorithms, applying robust software engineering practices, and exploring innovative research solutions. You'll work in a collaborative environment, contributing to the future of Machine Learning toolboxes at Apple.

The position offers competitive compensation ($143,100-$264,200) plus equity opportunities through stock programs. Benefits include comprehensive healthcare, retirement plans, education reimbursement, and potential relocation assistance. This is an excellent opportunity for those passionate about ML research and system engineering to impact products used by billions worldwide.

The ideal candidate brings 2+ years of experience in C++ and Python programming, strong GPU programming skills, and a solid understanding of Machine Learning algorithms. An advanced degree (M.S. or PhD) in Computer Science or related fields is required, along with excellent communication skills and experience with open-source development.