Sr. Software Engineer- AI/ML, AWS Neuron Apps

Benefits

Parental leave

Not verified not verified - source not recorded; timestamp not recorded

Non-birth-parent leave

Not verified not verified - source not recorded; timestamp not recorded

Family-building benefits

• Fertility benefits: Not verified
• Adoption assistance: Not verified
• Surrogacy assistance: Not verified

Mental health support

Not verified

Relocation assistance

Not verified

Childcare support

Not verified

Learning budget

Not verified

Verification

Not verified

Salary

Not verified not verified - source not recorded; timestamp not recorded

401(k) match

Not verified

Was this benefit information wrong? Tell us.

Schedule

Shift type

Not verified

Weekend work

Not verified

Application

Cover letter

Not verified

Assessment

Not verified

Deadline

Not stated

Where they hire

State eligibility is not yet verified.

About this role

Sr. Software Engineer- AI/ML, AWS Neuron Apps Seattle, Washington, USA Shape the Future of AI Accelerators at AWS Neuron Join the elite team behind AWS Neuron-the software stack powering AWS's next-generation AI accelerators Inferentia and Trainium. As a Senior Software Engineer in our Machine Learning Applications team, you'll be at the forefront of deploying and optimizing some of the world's most sophisticated AI models at unprecedented scale. What You'll Impact: • Pioneer distributed inference solutions for industry-leading LLMs such as GPT, Llama, Qwen • Optimize breakthrough language and vision generative AI models • Collaborate directly with silicon architects and compiler teams to push the boundaries of AI acceleration • Drive performance benchmarking and tuning that directly impacts millions of inference calls globally Key job responsibilities You will drive the Evolution of Distributed AI at AWS Neuron As a Technical Leader at the forefront of AWS's AI Accelerator, you'll architect the bridge between ML frameworks including PyTorch, JAX and AI hardware. This isn't just about just optimization-it's about revolutionizing how AI models run at scale. Technical Impact You'll Drive: • Spearhead distributed inference architecture for PyTorch and JAX using XLA • Engineer breakthrough performance optimizations for AWS Trainium and Inferentia • Develop ML tools to enhance LLM accuracy and efficiency • Transform complex tensor operations into highly optimized hardware implementations • Pioneer benchmarking methodologies that shape next-gen AI accelerator design What Makes This Role Unique: • Direct influence on AWS's AI infrastructure used by thousands of ML applications • Full-stack optimization

Read the full description at www.amazon.jobs. FewerJobs shows a source-linked preview and links to the original posting.