Software Development Engineer III, Annapurna Labs

What you'll do

• Research, design, build, test, deploy, and maintain software that improves Neuron workload porting and optimization.
• Lead the team’s technical direction and collaborate with Neuron teams, scientists, AWS AI Services, partners, and customers.
• Mentor engineers and provide technical guidance to improve team effectiveness.

What they ask for

• 5+ years leading system design or architecture, and 5+ years across the full software development life cycle.
• 5+ years of professional, non-internship software development and 5+ years programming in at least one language.
• Experience mentoring, serving as a tech lead, or leading an engineering team; hands-on Generative AI experience.
• Experience in at least one of: ML compilers, production coding agents, GenAI model architecture, model training, neural network optimization, or applied math.

Nice to have

• Master’s degree in computer science or equivalent; 2+ years in machine learning or computational modeling focused on hosting, building, or optimizing models for diverse hardware.
• Experience building AI agents for ML workload optimization or related tasks; open-source optimization community experience, AWS services knowledge, or ML domain knowledge.

Pay

184,900.00 - 250,200.00 USD annually (base salary)

Tools

AWS Neuron, Trainium, Generative AI, Machine Learning

AI focus

Builds AI agents, tools, and models to accelerate customer adoption and optimization of AWS Neuron and machine-learning workloads.