Senior AI/ML Engineer, Tools & Infrastructure jobs in United States
cer-icon
Apply on Employer Site
company-logo

Apple · 4 hours ago

Senior AI/ML Engineer, Tools & Infrastructure

Apple is a leading technology company, and they are seeking a Senior AI/ML Engineer to join their creative engineering team. This role involves leading the development of intelligent applications and systems that enhance hardware testing through data-driven insights, while also collaborating with various engineering teams to design robust AI/ML applications and implement production-grade AI pipelines.

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

Responsibilities

Architect and build robust, full-stack AI/ML applications using modern software engineering best practices, leveraging Python-based services (FastAPI), containerized deployments on Kubernetes, scalable data stores (Postgres/Timescale with vector search), and production-grade ML platforms for training, evaluation, and serving, while providing technical leadership across projects
Lead and mentor a team of external contractors, providing architectural guidance, code reviews, and daily direction to ensure fast, high-quality delivery
Drive the scope of engineering projects independently, identifying new opportunities, defining requirements, and managing the full project lifecycle from concept to deployment
Apply machine learning algorithms and statistical methods to solve complex business problems, focusing on rapid prototyping followed by high-quality production delivery
Design and implement MLOps workflows to ensure applications are scalable, monitoring their performance and reliability in production environments
Cultivate strong partnerships across engineering and business teams, acting as both a visionary and translator between software and non-software audiences
Act as the primary point of contact for stakeholders, tailoring communication to multidisciplinary audiences while delivering regular updates on project roadmaps
Present complex technical narratives and data-driven insights effectively to executive leadership through clear, high-impact presentations and visualizations

Qualification

AI EngineeringPythonMLOps workflowsMachine Learning algorithmsSoftware Engineering fundamentalsFastAPIKubernetesPostgres/TimescaleLeadershipCommunication skillsProblem-solving

Required

B.S. in Computer Science, Software Engineering, Computer Engineering, Machine Learning, or related field
4+ years experience and strong foundation in Software Engineering fundamentals, including data structures, algorithms, object-oriented design, and proficiency in building production-quality applications in Python
2+ years expertise in AI Engineering, using modern ML frameworks to design and deploy production-level platforms with end-to-end observability using vector-enabled data stores (e.g. Postgres/Timescale)
Demonstrated leadership experience with the ability to lead contractors, mentor peers, and manage technical resources effectively
Proven ability to drive projects independently: defining scope, collaborating with stakeholders, negotiating requirements, and driving projects to completion
Solid understanding of Machine Learning algorithms, with the ability to select and implement the right model or approach for the problem at hand
Excellent communication and presentation skills, with the ability to articulate complex technical concepts to diverse audiences and influence decision-making
Thrives in fast-paced, evolving environments—quickly pivots priorities while maintaining data integrity and reliability

Preferred

M.S. in Computer Science, Software Engineering, Computer Engineering, Machine Learning, or related field
Passion for quality and attention to detail; proactive in researching and assessing emerging technologies (AI/ML models, protocols, and techniques), and integrating them into production
Experience designing and operating MLOps workflows in cloud-native environments, including CI/CD for ML systems, model versioning and promotion, scalable batch and online inference, and monitoring of data quality, model performance, and system reliability in production
Track record of successfully growing the scope of engineering projects from initial proof-of-concept to organization-wide adoption
6+ years experience and strong foundation in Software Engineering fundamentals, including data structures, algorithms, object-oriented design, and proficiency in building production-quality applications
Experience with Python (FastAPI), PyTorch, Hugging Face, and Spark to deploy production-level platforms on Kubernetes with end-to-end observability, MLflow tracking, and vector-enabled data stores (e.g. Postgres/Timescale)
Experience with computer vision technologies and techniques, especially for segmentation, anomaly detection, and objective grading 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