Apple · 1 month ago
ML Engineer, FM Training Integration - ML Platform Technologies
San Francisco Bay Area
Full-time
Onsite
New Grad, Entry Level
$147K/yr - $221K/yrApple is a leading technology company, and they are seeking a ML Engineer to join their ML Compute team. The role involves improving the efficiency, scalability, and reliability of model training and inference workloads in the cloud, working closely with senior engineers and researchers.
AppsArtificial Intelligence (AI)BroadcastingDigital EntertainmentFoundational AIMedia and EntertainmentMobile DevicesOperating SystemsTVWearables
Responsibilities
Support the integration of model training and inference workloads with accelerator based cloud infrastructure
Assist with performance tuning of ML workloads to improve throughput, latency, and hardware utilization
Help identify and debug performance bottlenecks across data loading, model execution, and distributed training/inference
Collaborate with senior engineers to benchmark models and infrastructure configurations
Contribute to tooling, scripts, or pipelines that improve observability, reliability, and efficiency of ML workloads
Qualification
Required
Bachelor's degree in Computer Science, Engineering, or a related field
Basic understanding of machine learning workflows (training, evaluation, inference)
Familiarity with Python and at least one ML framework (e.g., PyTorch, TensorFlow, JAX)
Basic knowledge of cloud computing concepts (e.g., VMs, containers, storage, networking)
Interest in performance optimization, systems efficiency, and scalable ML infrastructure
Strong problem-solving skills and willingness to learn complex systems
Preferred
Exposure to GPU/TPU computing or accelerator-based workloads
Familiarity with distributed training or inference concepts (e.g., data parallelism, model parallelism)
Experience with containerization or orchestration tools (e.g., Docker, Kubernetes)
Basic understanding of profiling or benchmarking tools for ML workloads
Coursework or projects related to systems, cloud infrastructure, or performance engineering
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.
Founded in 1976
10001+ employees
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