Machine Learning Compiler Engineer
Description For Machine Learning Compiler Engineer
Apple's Neural Engine software team is at the forefront of enabling revolutionary Deep Learning technology across their product line, including Apple Vision Pro, iPhone, iPad, Apple Watch, and Mac. The team focuses on hardware acceleration using the Apple Neural Engine to achieve real-time, low-power, and high-performance execution of Deep Learning workloads. As a Machine Learning Compiler Engineer, you'll be instrumental in developing compiler technology to accelerate deep learning applications for Apple products. The role combines deep technical expertise in compiler architecture with the opportunity to work on cutting-edge AI hardware acceleration. You'll collaborate with hardware teams to optimize performance and power efficiency, while enabling both internal and third-party developers to leverage Apple's Neural Engine capabilities. The position offers competitive compensation, comprehensive benefits, and the chance to work on technology that impacts millions of users worldwide. This role is perfect for someone who combines compiler expertise with a passion for machine learning and hardware optimization, offering the opportunity to shape the future of AI acceleration in consumer devices.
Responsibilities For Machine Learning Compiler Engineer
- Architect and develop the compiler for Apple proprietary Neural Engine Accelerator architecture
- Bring up new hardware silicon and add support in the compiler for hardware features
- Work on bringing the compiler code to production quality
- Enable applications of deep learning technology for internal clients and 3rd party developers
- Evaluate existing hardware blocks and work with platform architecture team
- Work with micro-architecture design team to understand functional and performance goals
- Architect and lead complex compiler features
Requirements For Machine Learning Compiler Engineer
- In-depth working knowledge of compiler architecture, front-end and middle-end optimizations, scheduling, register allocation, back-end code generation
- Experience with IR and/or programming language design, esp. experience with MLIR and LLVM
- High level C++ programming expertise
- Experience working in a large and complex software stack
- Bachelor's degree in Computer Science, Computer Engineering or equivalent
- 10+ years relevant experience
Benefits For Machine Learning Compiler Engineer
- Comprehensive medical and dental coverage
- Retirement benefits
- Employee stock programs
- Discretionary restricted stock unit awards
- Employee Stock Purchase Plan with discount
- Educational expense reimbursement
- Discretionary bonuses
- Relocation assistance
- Range of discounted products and free services
