Lead Software Engineer - Data Platform ML (ASE)

Key Skills

Deep Learning, Design, Effective communication skills, GPUs, Infrastructure as code, Kubernetes-K8s, Large Language Models - LLMs, Leadership Skill, Machine learning techniques, Terraform

Job Description

You will be responsible for defining and driving the infrastructure roadmap for our data platform, offering the best automation, tooling and data security control across our stack at Apple scale. You will collaborate with cross functional teams of innovative software engineers, product managers, and engineering managers to ensure that GPU infrastructure is reliable, scalable and optimized for performance. We embrace the use of open source technologies including Kubernetes, Spark, Flink, Trino, Iceberg for data processing and Ray, Feature Platform for ML compute usecases.

RESPONSIBILITIES INCLUDE:

• Define and drive technical vision, roadmap, and strategy to manage GPU infrastructure of our platform for analytical and ML usecases.
• Participate in product design reviews to ensure performance optimization and monitoring is a core component of design
• Collaborate with stakeholders and cross-functional leaders in engineering, product, and operations across Apple to ensure the adoption of our data platform is done in a security compliant manner
• Liaison and coordinate with Corporate Information Security group for reviews, risk assessment, vulnerability treatment, security patches, etc
• Lead and mentor new hires or junior engineers
• Provide guidance and establish processes to ensure engineering excellence and operational sustainability with security compliance
• Foster a healthy, inclusive, collaborative, and technology-driven culture

Minimum Qualifications

• 10+ years of software development experience
• Experience with commercial and/or open source large scale data processing, storage frameworks and platforms
• Strong experience with infrastructure automation and provisioning including Kubernetes/Terraform.
• Experience architecting, building and operating large scale data processing systems in the public cloud
• Proficient in best practices and enforcement for data security, automation-driven, proactive monitoring
• Excellent verbal and written communication skills, able to collaborate cross-functionally with program managers and engineering partners
• Experience in influencing and driving key product innovations and opportunities across diverse collaborators
• BS, MS, or PhD degree in Computer Science or equivalent experience

Preferred Qualifications

• Familiarity with GPU computing or ML modeling frameworks strongly preferred.
• Experience working with or developing Large-language models (LLMs)
• Experience developing and optimizing algorithms that run efficiently on resource constrained platforms
• Design, implementation and benchmarking/fine-tuning of ML/deep learning algorithms
• Experience with observability tools like Prometheus and Grafana.

Pay & Benefits

At Apple, base pay is one part of our total compensation package and is determined within a range. This provides the opportunity to progress as you grow and develop within a role. The base pay range for this role is between $175,800 and $312,200, and your base pay will depend on your skills, qualifications, experience, and location.

Apple employees also have the opportunity to become an Apple shareholder through participation in Apple’s discretionary employee stock programs. Apple employees are eligible for discretionary restricted stock unit awards, and can purchase Apple stock at a discount if voluntarily participating in Apple’s Employee Stock Purchase Plan. You’ll also receive benefits including: Comprehensive medical and dental coverage, retirement benefits, a range of discounted products and free services, and for formal education related to advancing your career at Apple, reimbursement for certain educational expenses — including tuition. Additionally, this role might be eligible for discretionary bonuses or commission payments as well as relocation. Learn more about Apple Benefits.

Note: Apple benefit, compensation and employee stock programs are subject to eligibility requirements and other terms of the applicable plan or program.