AIML - Senior Applied ML Engineer, Health AI

Key Skills

JAX framework, Large Language Models - LLMs, Natural Language Processing (NLP), PyTorch, Swift, TensorFlow

Job Description

The Health AI team is at the forefront of machine learning and health science at Apple. We are a close-knit team of highly accomplished, deeply technical research scientists and machine learning engineers focused on delivering innovative technologies that impact millions of users. We are looking for an applied ML engineer passionate about solving real-world problems in the health domain that make a difference in our customers' lives.

Key Qualifications

• Hands-on experience in state-of-the-art machine learning and deep learning as applied to large-scale datasets
• Strong background in generative models, natural language processing, and large language models
• Passionate about applying machine learning methods to deliver end-user experiences in health space
• Proficiency developing large scale models using modern machine learning packages (e.g. TensorFlow, PyTorch, Jax)
• Strong engineering skills and experience in writing production-quality code in Python, Swift or other programming language
• Experience of prompt engineering, fine-tuning, evaluating, and developing data collection/annotation/management tooling for large language models.
• Experience with one of dialog and conversational systems, information retrieval, data mining
• Excellent interpersonal and communication skills

Description

In this role, you will use your skills and experience in state-of-the-art machine learning and deep learning to design, implement and evaluate new machine learning models and algorithms to solve challenging problems involving unique data and objectives. You will have the opportunity to explore and develop generative AI technologies. The ability to substantially improve model performance from a prototype by leveraging novel architectures, optimizing hyper-parameters or exploring data augmentation techniques is key. The successful candidate should also possess strong engineering skills and the ability to set up flexible data pipelines, modeling, evaluation and analysis platforms to successfully transform research prototypes to production quality features.

Education & Experience

BS, MS or Ph.D. in Computer Science, Machine Learning, or related fields or an equivalent qualification obtained through other avenues

Pay & Benefits

At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $161,700 and $284,900, and your base pay will depend on your skills, qualifications, experience, and location.

Apple employees also have the opportunity to become an Apple shareholder through participation in Apple’s discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan. You’ll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses — including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. Learn more about Apple Benefits.

Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.