AI/ML - ML/HW/SW Co-Design Engineer, DMLI
The Data and Machine Learning Innovation (DMLI) team employs ML/HW/SW co-design to achieve best-in-class performance and energy efficiency for numerous use cases that deploy neural networks. We seek a generalist to help define and implement features that would be realized in next-gen neural accelerators. Our team is comprised of HW, SW, and ML engineers working together in the area of Efficient ML. Our charter is to push the frontiers of performance and power for AI inference.
As a co-design engineer, you will be accountable for bridging the gap between ML and devices. This includes working with ML, computer architecture, and SW teams to understand the trade-offs from various viewpoints when it comes to ML inference, keeping abreast of advancements in the ML efficiency techniques space particularly for Generative AI, and coming up with novel solutions that address the ever-growing needs for future ML workloads.
On a day-to-day basis, your responsibilities include but are not limited to:
- Architecting energy-efficient hardware designs
- Implementing changes in an in-house simulator
- Implementing new features in an in-house compiler
- Working with ML engineers to optimize their networks for HW acceleration
Our ideal team member is courageous when it comes to trying new things, is adept at reasoning about systems performance, and is willing to iterate on ideas. We value team members with strong communication skills and experience working cross-functionally with HW, SW, and ML teams.
Minimum Qualifications:
- Bachelor's, Master's, or PhD degree in Electrical Engineering, Computer Science, or a related field
- 5+ years of experience working on ML inference performance optimization either in SW or HW space
- Proficient in Python, working knowledge of C++
- Proficiency in algorithms and coding skills to optimize a workload and solve hard problems in scheduling, allocation, and tune solutions to the details of an architecture platform
Preferred Qualifications:
- Deep understanding of computer systems and the interactions between HW and SW
- Strong communicator with ability to analyze complex and ambiguous problems
- Experience with backend compilation, ML/HW/SW co-design, and/or performance optimization
- Familiarity with ML model efficiency techniques (e.g., quantization, pruning) and their bottlenecks at inference time
Apple is an equal opportunity employer that is committed to inclusion and diversity. We take affirmative action to ensure equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics. Learn more about your EEO rights as an applicant.
#J-18808-Ljbffr