Software Development Engineer, AI/ML, AWS Neuron, Model Inference

Job overview

The Software Development Engineer, AI/ML, AWS Neuron, Model Inference role is an onsite position based in Cupertino, CA. Compensation is not specified, though the role supports the Inference Enablement and Acceleration team. This role exists to build distributed inference support for PyTorch in the Neuron SDK. The engineer will tune models to maximize efficiency on Trainium and Inferentia silicon, working across the stack from system-level optimizations to framework-level integration.

What we're looking for

Skills & competencies

full-time

day shift

cupertino ca

annapurna labs

software engineer

ai/ml

model inference

aws neuron

python

c++

onsite

no on-call

no travel

Work arrangement

• Weekend coverage required

Benefits & perks

• Health insurance, 401(k) matching, paid time off, parental leave, sign-on payments, restricted stock units (RSUs)

About the employer

Annapurna Labs (U.S.) Inc. is hiring for this role. Industry: Custom Computer Programming Services. Sector: 54.