AIML - Machine Learning SW/HW Co-Design Engineer, Machine Learning Platform & Infrastructure

$175,800 - $312,200/Yr

Apple - Cupertino, CA

Full-time - Mid Level

Cupertino, CA

Computer and Electronic Product Manufacturing

About the position

The Machine Learning and Platforms (MLPT) team at Apple is focused on building the inference stack for on-device machine learning (ML) that operates on Apple Silicon. This role involves developing converters and compilers for translating network definitions, optimizing ML networks, and managing execution on hardware. The team collaborates with various internal teams and external app developers to enhance ML performance across a range of devices, from wearables to Macs.

Responsibilities

• Develop converters and compilers for ML network definitions.
• Optimize ML networks for performance and efficiency.
• Manage execution of ML tasks on hardware.
• Collaborate with cross-functional teams to improve hardware/software co-design.
• Conduct deep analysis of software and hardware for innovative improvements.
• Prototype new approaches to enhance inference without sacrificing accuracy.
• Perform comprehensive analyses on performance, power, and accuracy of ML models.

Requirements

• Understanding of machine learning fundamentals and current research.
• Experience with training frameworks like PyTorch, TensorFlow, or JAX.
• Knowledge of computer architecture (CPU and GPU) and performance modeling.
• Experience with ML systems, particularly for on-device inference.
• Ability to perform detailed analyses starting from first principles of deep learning techniques.
• Strong programming skills in C/C++ and/or Python.
• Excellent communication skills and a collaborative mindset.

Nice-to-haves

• Masters or PhD in Computer Science, Computer Science & Engineering, or Computer Engineering.

Benefits

• Comprehensive medical and dental coverage
• Retirement benefits
• Discounted products and free services
• Tuition reimbursement for formal education
• Discretionary bonuses or commission payments
• Relocation assistance
• Employee stock purchase plan participation