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

Key Skills

3D data processing, Aartificial intelligence, Algorithms, Github, JAX framework, Machine learning techniques, Natural Language Processing (NLP), Optimization, Python Programming, PyTorch, Streaming APIs

Job Description

Key Qualifications

• Expertise with Python
• 4+ years validated experience
• Consistent track record of delivering and maintaining high quality production software
• Demonstrated ability to design user friendly and maintainable APIs
• Excellent technical communications skills
• Ability to mentor and lead execution of complex features via planning, effective delegation and continuous tracking across multiple sprint cycles
• Ability to collaborate 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
• Expertise with PyTorch, TF or JAX (preferred)
• Experience with model compression and quantization (preferred)

Description

We develop, prototype and productize groundbreaking algorithms for neural network model compression. Our algorithms are implemented using PyTorch, and optimizations are geared towards efficient deployment via Core ML. We optimize models across domains, including NLP, vision, text and generative models. Our APIs are available to Core ML users, both internal to Apple and external developers via the Core ML Tools optimization sub module. As a successful engineer on our team, you will: - Implement the latest algorithms from research papers for model compression - Set up training jobs, datasets, evaluation, performance benchmarking pipelines - Run detailed experiments to profile algorithms on various models and across different sizes and maintaining 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 sources such as GitHub - Expand documentation with examples, benchmark data to drive user adoption and address user problems - Self prioritize and adjust to changing priorities and asks - Be a phenomenal teammate who enjoys doing things that make the overall project successful and others in the team excel. This includes working on detailed reviews of PRs and technical docs, bug fixes, test infra etc.

Education & Experience

BS/MS/PhD in Computer Science or Electrical Engineering