Senior Machine Learning Engineer jobs in United States
info-icon
This job has closed.
company-logo

Apple · 8 hours ago

Senior Machine Learning Engineer

Apple is a place where extraordinary people gather to do their lives best work. They are seeking a Senior Machine Learning Engineer to research and develop advanced machine learning solutions for various applications, work closely with product teams, and provide technical guidance to improve workflows for model development and deployment.

AppsArtificial Intelligence (AI)BroadcastingDigital EntertainmentFoundational AIMedia and EntertainmentMobile DevicesOperating SystemsTVWearables
check
Comp. & Benefits
check
H1B Sponsor Likelynote

Responsibilities

Research and develop advanced machine learning solutions for natural language processing (NLP), audio, computer vision, and multi-modal applications
Work closely with crossfunctional partners, including product teams, to productionize large-scale ML solutions
Provide technical guidance to improve workflows for training, evaluation, model optimization, and deployment
Contribute modular and performant Python code to internal PyTorch model development frameworks; develop, release, and support new features for these projects
Implement and maintain state-of-the-art neural networks architectures in internal frameworks
Architectures include large language models (LLMs), audio models, vision transformers, and multimodal generative AI models
Design, implement, and maintain production workflows for large-scale distributed training using advanced techniques such as fully sharded data parallelism and model parallelism
Implement and support tooling for evaluating the quality of large deep learning models
Research and evaluate the latest techniques for machine learning model development; incorporate them into production workflows
Contribute to the organization’s technical roadmap by identifying areas for innovation in machine learning solutions
Provide technical leadership to improve workflows for training, evaluation, model optimization, and deployment
Mentor and guide junior engineers and interns in best practices for machine learning model development

Qualification

Machine LearningNatural Language ProcessingComputer VisionPythonPyTorchJAXDistributed ML FrameworksModel ParallelismFine-tuning TechniquesEvaluation FrameworksData StructuresAlgorithmsGitGitHub

Required

Master's degree or foreign equivalent in Computer Science, Engineering, or related field and 3 years of experience in the job offered or related occupation
2 years of experience with each of the following skills is required:
Applying ML fundamentals to make quantifiable improvements to prior solutions in the domains of ML systems, natural language processing, computer vision, and/or audio processing
Utilizing modern ML frameworks, including PyTorch or JAX, to train and evaluate generative AI models for natural language processing, computer vision, or audio applications
Applying software engineering fundamentals, with a strong focus on data structures and algorithms, to develop efficient and maintainable Python code for ML applications
Using collaboration tools including git and GitHub to contribute to a large codebase with 3+ collaborators
Implementing and maintaining tooling for industrial ML model development using Python and PyTorch or JAX
Implementing state-of-the-art ML model architectures in PyTorch or JAX, including transformer models for text, vision, or audio applications
Applying fundamentals of ML systems to scale up ML training to large models (at least 1 billion parameters) using fully sharded data parallelism (FSDP) or model parallelism
Applying parameter efficient fine-tuning techniques, including LoRA (low-rank adaptation), to fine tune foundation models in a computationally efficient way
Using distributed machine learning frameworks, including torch.distributed, DeepSpeed, or Ray, to build machine learning infrastructure or platforms
Evaluating the quality of generative AI models using evaluation frameworks including lm-evaluation-harness

Benefits

Comprehensive medical and dental coverage
Retirement benefits
A range of discounted products and free services
Reimbursement for certain educational expenses — including tuition
Discretionary bonuses or commission payments
Relocation

Company

Apple is a technology company that designs, manufactures, and markets consumer electronics, personal computers, and software.

H1B Sponsorship

Apple has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (6998)
2024 (3766)
2023 (3939)
2022 (4822)
2021 (4060)
2020 (3656)

Funding

Current Stage
Public Company
Total Funding
$5.67B
Key Investors
Berkshire HathawayMicrosoftSequoia Capital
2025-05-05Post Ipo Debt· $4.5B
2025-01-16Post Ipo Debt· $0.31M
2021-04-30Post Ipo Equity

Leadership Team

leader-logo
Tim Cook
CEO
leader-logo
Craig Federighi
SVP, Software Engineering
Company data provided by crunchbase