ML Infrastructure Engineer - Multimodal Training Tools, SIML jobs in United States
cer-icon
Apply on Employer Site
company-logo

Apple · 1 month ago

ML Infrastructure Engineer - Multimodal Training Tools, SIML

Apple is seeking engineers experienced in building tools for training, adapting, and deploying large-scale generative models. The role involves working with a cross-functional team to develop ML infrastructure and algorithms, focusing on enhancing generative modeling technologies.

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

Responsibilities

Contributing towards tools for large generative model training including diffusion & autoregressive workflows
Tools for efficient inference and hosting of models for experimentation and human feedback
Tooling for model representation and efficient deployment on multiple HW targets incl. Apple Silicon
Benchmarking, Analysing and Improving training and inference performance
Integrating efficient data loading strategies and auto-eval workflows
CI/CD of base training workstreams

Qualification

ML toolingPyTorchLLM trainingDistributed trainingML FundamentalsCross-functional teamwork

Required

Bachelors, Masters, or PhD in Electrical Engineering/Computer Science or a related field (mathematics, physics or computer engineering), with a focus on machine learning; or comparable professional experience
Experienced in training / adapting LLM and Diffusion models
Advanced Fluency in PyTorch
Excellent programming skills and experience contributing software to large projects
Experience with distributed training of large models

Preferred

Strong ML Fundamentals
Experience working with large cross-functional and diverse teams

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