Software Engineer - AML, AI & Data Platforms (AiDP)

Apple • Sunnyvale, CA

2d

About The Position

AI & Data Platforms (AiDP) is IS&T's engine for AI-powered innovation. The team brings together data, application development, and machine learning — including generative AI — along with data services and customer success functions, to help IS&T build solutions more efficiently and streamline the adoption and embedding of generative AI across Apple. We are looking for a passionate and experienced Software Engineer to help build the next generation of Applied Machine Learning Platform. The Applied Machine Learning Platform team provides backend services and infrastructure for various Machine Learning and Data Science teams to train, build, deploy and inference models at scale to prevent Fraud on multiple Apple Platforms like Apple Pay, Apple Media Products, App Store, Online Store, Retail, AppleCare and Manufacturing. In addition to preventing Fraud, this platform is responsible for driving Operations and Logistics for Online Store, AppleCare and Retail. Our team within the greater AiDP Platform team is the Core Services, which is the backbone of the platform, responsible for handling thousands of transactions per second in a distributed manner. As a Software Engineer, you will have deep systems thinking to design, build, and enhance scalable and highly concurrent ML and AI serving platforms. Knowledge of Python and Java, Machine Learning concepts, tools, and packages is a must. If you’re excited about building production-grade platforms and solving hard distributed systems problems, this is your opportunity to make a lasting impact at scale. You will design and implement large-scale, secure, and highly available systems, while collaborating across teams to drive the future of secure scalable inference platforms. The mindset required and to be developed is how to process thousands of transactions per second, and how to achieve consistency without sacrificing performance. Work with cross-functional teams to drive requirements, size scope and effort, mentor junior engineers, lead projects to completion, and provide support for any production issues.

Requirements

• 6+ years of software development experience
• 4+ years of Python or Java expertise building Production Applications
• Solid understanding of concurrency, threading, multiprocessing and their practical trade-offs in high-throughput production systems
• Strong understanding of language internals and their implications in highly concurrent systems
• Deep knowledge of ML and AI frameworks and ecosystem
• Strong understanding of computer science fundamentals, including data structures and algorithms
• Proficiency in Containerization technologies like Kubernetes and CNCF technologies
• Proficiency in profiling, optimizing, debugging applications and libraries
• Bachelor’s Science in Computer Science or similar degree or discipline or experience
• Excellent written and verbal communication skills on both technical and non-technical topics
• Ability to debug complex cross-platform issues and optimize performance
• Ability to debug and improve ML workflow tuning for high accuracy
• Self-directed, self-motivated and detail oriented with ability to come up with good design proposals and thorough analysis of production issues

Nice To Haves

• Exposure to Ray, KServe, Triton Or other inference frameworks
• Knowledge of using GenAI to build solutions to improve productivity of developers, users, and business

Responsibilities

• Design, build, and enhance scalable and highly concurrent ML and AI serving platforms.
• Design and implement large-scale, secure, and highly available systems.
• Collaborate across teams to drive the future of secure scalable inference platforms.
• Process thousands of transactions per second and achieve consistency without sacrificing performance.
• Work with cross-functional teams to drive requirements, size scope and effort.
• Mentor junior engineers.
• Lead projects to completion.
• Provide support for any production issues.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior