Machine Learning Ops, Applied Machine Learning

Posted: 20 Apr 2024

Weekly Hours: 40

Role Number:200545290

Apple’s Applied Machine Learning team has built systems for a number of large-scale data science applications. We work on many high-impact projects that serve various Apple lines of business. We use the latest in open source technology and as committers on some of these projects, we are pushing the envelope. Working with multiple lines of business we manage many streams of Apple-scale data. We bring it all together and unleash business value. We do all this with an exceptional group of software engineers, data scientists, DevOps engineers and managers. We are looking for a talented and dedicated engineers to join our team to bring passion for infrastructure and distributed systems, to build world-class platforms/products at a very large scale across cloud environments.

Key Qualifications

4+ years of experience in Python Programming
Extensive experience in deploying and managing the applications on AWS/GCP & Kubernetes
Deep understanding on RAG based pipelines for Model inferencing, GuardRails
Experience in open source LLM App frameworks like LangChain/LlamaIndex
Exposure to Cloud managed services like AWS BedRock/GCP Vertex AI
Good Understanding of Agents in GenAI
Strong Experience in Infrastructure templating tools like CloudFormation, Terraform
Experience in GitOps based deployment tools like Spinnaker/Flux/ArgoCD
Strong proficiency with Helm and Kustomize for managing Kubernetes applications and configurations.
Experience in managing Embeddings using Vector databases
Exposure to Promot engineering
Experience in observability & traceability for Large Language Models.
Experience in Performance tuning on operating systems like Linux
Excellent analytical & problem solving skills

Description

Join Apple's Applied Machine Learning Team, as a Senior Software Engineer, to enable GenAI across our Applications & Platforms. Candidates should have strong background in LLM core concepts, be proficient in setting up and supporting the large scale big data applications in public cloud like AWS/GCP. The main responsibilities for this position include: Build LLM Applications using open source LLM App Frameworks, AWS BedRock/GCP Vertex AI Evaluate and port Language Models onto optimized infrastructure to reduce cost and increase performance Build tools to benchmark and compare various embedding databases, LLMs Build & Support CI/CD tools to port & manage applications on AWS/GCP & Kubernetes Build automation to enable self-healing systems Ability to troubleshoot application specific, core network, system & performance issues. Build a multi-tenancy system by enforcing data protection between different use cases. Involvement in challenging and fast paced projects supporting Apple's business by delivering innovative solutions. The candidate is expected to be self-motivated, proactive, and a solution-oriented individual.

Education & Experience

BS in computer science with 4+ years or MS plus 2+ years experience or related experience.

Additional Requirements

Exposure to various LLM infrastructure like GPUs, TPUs & Inferentia is preferred
Exposure to LLM runtime like Triton, Frameworks like TensorRT, vLLM is an added advantage.
Exposure to general Java troubleshooting skills

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 $138,900.00 and $256,500.00, 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.

Apple is an equal opportunity employer that is committed to inclusion and diversity. We take affirmative action to ensure equal opportunity for all applicants without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, Veteran status, or other legally protected characteristics.