AIML - Senior Software Engineer, Model Optimization Library, On-Device Machine Learning

$143,100 - $264,200/Yr

Apple - Cupertino, CA

Full-time - Mid Level

Cupertino, CA

Computer and Electronic Product Manufacturing

About the position

The Senior Software Engineer for the Model Optimization Library focuses on developing and implementing advanced algorithms for compressing deep learning models to enhance user experiences on Apple devices. This role involves creating a Python library that provides easy-to-use APIs for model optimization techniques such as quantization and pruning, ensuring compatibility with Core ML and Apple hardware. The engineer will collaborate with various teams to drive the adoption of these optimizations across a wide range of applications.

Responsibilities

• Implement latest algorithms from research papers for model compression
• Set up training jobs, datasets, evaluation, and performance benchmarking pipelines
• Run detailed experiments to profile algorithms on various models and across different sizes
• Maintain model cards and documentation
• Collaborate with ML practitioners to co-develop algorithms customized for Apple hardware
• Manage Python releases and ensure API backward compatibility
• Provide user support via various channels including open source GitHub
• Expand documentation with examples and benchmark data to drive user adoption
• Self-prioritize and adjust to changing priorities
• Conduct detailed reviews of PRs and technical documents, bug fixes, and test infrastructure.

Requirements

• Proven experience in developing and implementing machine learning algorithms
• Strong proficiency in Python and experience with PyTorch
• Experience with model optimization techniques such as quantization and pruning
• Ability to set up and manage training jobs and performance benchmarking
• Experience in collaborating with cross-functional teams
• Strong problem-solving skills and ability to adapt to changing priorities.

Nice-to-haves

• Experience with Core ML and its optimization tools
• Familiarity with neural network architectures across NLP, vision, and generative models
• Experience in user support and community engagement
• Knowledge of software release management and API design.

Benefits

• Comprehensive medical and dental coverage
• Retirement benefits
• Discounted products and free services
• Reimbursement for certain educational expenses including tuition
• Discretionary bonuses or commission payments
• Relocation assistance
• Participation in employee stock programs.