Foundation Models ML Engineer — Production-Ready AI

Apple Inc.; New York (NY); On-site; USD 181,100 - 318,400; Full time

Benefits offered by this job

Comprehensive medical and dental coverage

Retirement benefits

Employee stock purchase plan

Educational reimbursement

Job summary

A leading technology company in New York City seeks a Machine Learning Engineer to design and implement scalable machine learning systems. The ideal candidate will have a strong background in deep learning techniques, model optimization, and AI frameworks. This role offers an annual salary range between $181,100 and $318,400, along with comprehensive benefits including medical coverage, stock options, and educational reimbursements.

Qualifications

• MS or PhD in relevant technical field required.
• Expert programming skills in Python expected.
• Experience with large-scale model training and deployment necessary.

Responsibilities

• Design scalable machine learning infrastructure for foundation models.
• Collaborate with teams to productionize AI algorithms.
• Optimize models for performance and efficiency.

Skills

Expert-level programming skills in Python

Proficiency in machine learning frameworks such as Jax, PyTorch, TensorFlow

Experience with large-scale model training and deployment

Familiarity with Kubernetes, Docker, Cloud platforms (AWS, GCP, Azure)

Strong background in: Distributed training, Model optimization

Education

MS or PhD in Computer Science, Machine Learning, or related technical field

Tools

Jax

PyTorch

TensorFlow

Kubernetes

Docker

Job description

A leading technology company in New York City seeks a Machine Learning Engineer to design and implement scalable machine learning systems. The ideal candidate will have a strong background in deep learning techniques, model optimization, and AI frameworks. This role offers an annual salary range between $181,100 and $318,400, along with comprehensive benefits including medical coverage, stock options, and educational reimbursements.