ML Systems Engineer for GenAI & Neuron Tools

Amazon Web Services (AWS); New York (NY); On-site; USD 158,000 - 214,000; Full time

Job summary

Amazon Annapurna Labs in New York seeks a talented software engineer to push the GenAI revolution by building agents, tools, and models that streamline adoption of AWS Neuron, the ML silicon software stack. You will partner with internal and external customers to identify obstacles and accelerate migration to the company’s ML hardware.

You will design and implement high-quality software, solve complex problems, and contribute to scalable solutions across compiler, hardware, and ML engineering

Qualifications

• 3+ years of non-internship professional software development experience.
• 2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems.
• Experience programming with at least one software programming language.
• CS core: object-oriented design, data structures, and performance analysis with at least 2 programming languages.
• Experience in ML compilers, production coding agents, GenAI model architecture, model training, neural network optimization, or related applied math.

Responsibilities

• Research implementations that deliver the best possible experiences for customers.
• Deliver on goals to improve the time and effort it takes to port and optimize ML workloads on Neuron.
• Solve challenging technical problems across all stack layers.
• Design, implement, test, deploy and maintain software solutions to improve performance, durability, cost, and security.
• Build high-quality, highly available products.
• Potentially contribute intellectual property through patents.

Skills

Software development

System design

Programming languages

CS fundamentals

ML compilers

Tools

AWS Neuron

Open-source ML frameworks

Job description

Amazon Annapurna Labs in New York seeks a talented software engineer to push the GenAI revolution by building agents, tools, and models that streamline adoption of AWS Neuron, the ML silicon software stack. You will partner with internal and external customers to identify obstacles and accelerate migration to the company’s ML hardware.

You will design and implement high-quality software, solve complex problems, and contribute to scalable solutions across compiler, hardware, and ML engineering