AIML - Applied Machine Learning Engineer, MIND
As a Machine Learning Engineer in the Machine Intelligence Neural Design (MIND) team at Apple, you will be part of the Data and Machine Learning Innovation (DMLI) organization. With roots in ML, computer vision, and energy efficiency research, MIND is strategically positioned to contribute to both short-term and long-term projects, from well-known Apple products to ambitious, high-risk, high-reward initiatives.
This role focuses on shipping ML-based features and products. You will innovate in the entire end-to-end ML production pipeline, crafting creative approaches to datasets, model training, and on-device inference optimizations. We value fearless team members who are willing to try new things, iterate quickly, and prototype ideas from start to finish, resulting in high-quality implementations.
You will collaborate closely with hardware, computer architecture, and software teams to drive data-driven decision-making and innovation. Your daily responsibilities will include analyzing and leveraging large datasets, identifying and addressing data gaps, predicting potential future gaps, and designing robust data collection strategies. Additionally, you will build, fine tune, and optimize machine learning models, ensuring their performance is fine-tuned for on-device deployment.
As a team member, you will leverage your background to:
- Develop features and models enhancing machine learning systems. This includes scaling up model training, building data pipelines, and optimizing performance.
- Review and implement cutting-edge machine learning algorithms.
- Work closely with cross-functional partners across the stack, including; Apps, Compilation, Hardware and etc.
- Proficient in Python and deep learning frameworks (e.g., PyTorch).
- Experience training ML models, including deep learning models.
- Expertise in large-scale data processing, distributed computing, and data engineering.
- Able to define metrics, evaluate ML models, and conduct error analysis.
- Bachelor's, Master's, or PhD in Computer Science or a related field.
- Experience shipping ML features and products.
- Familiar with cloud platforms (AWS, GCP).
- Experience modeling vision problems in object detection, facial recognition, and/or temporal machine learning.
- Experience building efficient ML models for on-device deployment
- Strong communicator with the ability to analyze complex and ambiguous problems.
- Familiar with ML model efficiency techniques (e.g., quantization, pruning) and their inference-time bottlenecks.
- Knowledge of recent advances in deep learning and computer vision.
- Proven track record of contributing to diverse teams in a collaborative environment
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.