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
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
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
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 CompanyTotal Funding
$5.67BKey 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
Tim Cook
CEO
Craig Federighi
SVP, Software Engineering
Recent News
Venrock
2025-12-01
2025-09-25
Mac Daily News
2025-09-25
Company data provided by crunchbase