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

$143,100 - $264,200/Yr

Apple - Monte Vista, CA

Full-time - Mid Level

Monte Vista, CA

Computer and Electronic Product Manufacturing

About the position

The role focuses on developing and optimizing machine learning model compression techniques to enhance user experiences across Apple devices. The team is responsible for creating a Python library that implements various quantization and pruning algorithms, making them accessible to developers through user-friendly APIs. The position involves collaboration with multiple teams within Apple to ensure seamless integration with Core ML and efficient deployment of models.

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 maintain model cards
• Collaborate with ML practitioners across the company to co-develop and implement algorithms customized for Apple hardware
• Manage Python releases and API backward compatibility
• Provide user support via various channels including engagement via open source GitHub
• Expand documentation with examples and benchmark data to drive user adoption and address user pain points
• Self-prioritize and adjust to changing priorities and asks
• Conduct detailed reviews of PRs and technical docs, bug fixes, and test infrastructure.

Requirements

• Expertise with Python
• 4+ years industry experience
• Proven track record of delivering and maintaining high quality production software
• Demonstrated ability to design user-friendly and maintainable APIs
• Excellent technical communication skills
• Proven track record of mentoring and leading execution of complex features via planning, effective delegation, and continuous tracking across multiple sprint cycles
• Ability to collaborate effectively and excel in projects involving cross-functional teams
• Knowledge of machine learning pipeline, from data set up, data processing, training models, and evaluation
• Demonstrated ability to learn quickly and grasp ML concepts, architectures, and optimization algorithms.

Nice-to-haves

• Expertise with PyTorch, TF, or JAX
• Experience with model compression and quantization
• BS/MS/PhD in Computer Science or Electrical Engineering

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