Be an early applicantLess than 25 applicants

Company

Original Job Post

Apple · 3 days ago

ML Engineer - Machine Learning Platform & Infra

Seattle, WA

Full-time

Onsite

Senior Level

$162K/yr - $285K/yr

5+ years exp

Wonder how qualified you are to the job?

Maximize your interview chances

Consumer ElectronicsHardware

H1B Sponsorship

Comp. & Benefits

Insider Connection @Apple

Discover valuable connections within the company who might provide insights and potential referrals, giving your job application an inside edge.

Responsibilities

Work alongside Foundation Model Research team to optimize inference for cutting edge model architectures.

Work closely with product teams to build production-grade solutions to launch models serving millions of customers in real-time.

Build tools to understand bottlenecks in Inference for different hardware and use cases.

Mentor and guide engineers in the organization.

Qualification

Find out how your skills align with this job's requirements. If anything seems off, you can easily click on the tags to select or unselect skills to reflect your actual expertise.

Project managementCloud computingKubernetesDockerGpu programmingCudaMachine learningPytorchTensorflowLeadershipProblem-solvingGolangPythonDeep LearningTransformersEncoder/Decoder modelsNvidia TensorRT-LLMNvidia vLLLMDeepSpeedNvidia Triton ServerOpenAI Triton

Required

5+ years of experience leading and driving complex, ambiguous projects.

Have experience with high throughput services particularly at supercomputing scale.

Proficient with running applications on Cloud (AWS / Azure or equivalent) using Kubernetes, Docker etc.

Familiar with GPU programming concepts using CUDA.

Familiar with one of the popular ML Frameworks like Pytorch, Tensorflow.

Preferred

Proficient in building and maintaining systems written in modern languages (eg: Golang, python)

Familiar with fundamental Deep Learning architectures such as Transformers, Encoder/Decoder models.

Familiarity with Nvidia TensorRT-LLM, vLLLM, DeepSpeed, Nvidia Triton Server etc.

Experience writing custom CUDA kernels using CUDA or OpenAI Triton.

Benefits

Comprehensive medical and dental coverage

Retirement benefits

Discounted products and free services

Reimbursement for certain educational expenses

Possibility of discretionary bonuses or commission payments

Possibility of relocation assistance

Company

Apple

Glassdoor

4.2

Apple is a corporation that designs, manufactures, and markets consumer electronics, personal computers, and software.

Founded in 1976

Cupertino, California, USA

10,001+ employees

http://www.apple.com

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

Trends of Total Sponsorships

2023 (2569)

2022 (4857)

2021 (4908)

2020 (3580)