AIML - ML Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Apple · 19 hours ago

AIML - ML Engineer

Apple is a leading technology company known for its innovative products and services. They are seeking a highly motivated ML Engineer to work with researchers on building high-performance software for novel machine learning algorithms, with a focus on speech processing and scalable solutions.

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

Responsibilities

Work with researchers on the team to build high-performance and scalable software addressing novel ML research algorithms
Apply solid software engineering skills, leverage experience to deal with the unexpected, explore research software solutions and pave the way to future Machine Learning toolboxes
Be part of a small team dedicated to advancing ML algorithms and techniques - Is this you? If so we'd love to hear from you

Qualification

Python programmingML framework expertiseML benchmarks experienceSpeech processingCUDA programmingHigh-Performance ComputingDistributed computingCollaborative software developmentProblem-solving readinessC++ programmingMLX experienceCommunication

Required

M.S. in Computer Science (or related fields) or equivalent experience
Expert in Python programming and tensor operations, including https://github.com/rougier/numpy-100
Expert in at least one ML framework (PyTorch preferred but tensorflow or jax are ok as well), including flash-attention, efficient kv-caching, DDP and FSDP
Hands-on experience in establishing ML benchmarks (data, models, metrics)
Involvement with open-source projects and experience with collaborative software development
Excellent communication skills
Readiness to encounter unforeseen challenges and to solve them

Preferred

Speech processing (including generative modeling of speech) experience is a plus
PhD in Computer Science (or related fields) or equivalent experience
Expert in Python programming and tensor operations, including https://github.com/rougier/numpy-100
Expert in at least one ML framework (PyTorch preferred but tensorflow or jax are ok as well), including flash-attention, efficient kv-caching, DDP and FSDP
Hands-on experience in establishing ML benchmarks (data, models, metrics)
Active development in open-source projects and experience with collaborative software development
Experience with CUDA programming, and/or High-Performance Computing and/or distributed computing
Excellent communication skills
Readiness to encounter unforeseen challenges and to solve them
Past research experience
C++ programming is a plus
MLX experience is a plus

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