Senior AI Inference SDK Lead (Trainium/Inferentia)

Amazon Web Services (AWS); Cupertino (CA); On-site; USD 253,000 - 342,000; Full time

Benefits offered by this job

Stock options / RSUs

Health insurance

401(k) matching

Paid time off

Parental leave

Job summary

Annapurna Labs (U.S.) Inc. in Cupertino, CA is seeking a Sr. Software Development Manager to lead a team developing components and libraries enabling AI model optimization on Trainium and Inferentia accelerators within the AWS Neuron SDK.

You will collaborate with ML model developers and cross-functional teams to deliver high-performance, usable SDK capabilities and shape the strategic direction with senior leadership.

Qualifications

• 10+ years of engineering experience.
• 5+ years of engineering team management experience.
• 10+ years of software development experience.
• Experience partnering with product or program management teams.
• Experience managing multiple concurrent programs, projects and development teams in an Agile environment.

Responsibilities

• Responsible for the full life cycle of developing and releasing components of the AWS Neuron SDK to support the Trainium and Inferentia accelerators.
• Develop and grow and adapt your team to meet the needs of the rapidly changing AI software ecosystem.
• You will work with the executive leadership and other senior management and technical leaders to define strategic directions and deliver new capabilities to ML model developers and users.

Skills

Engineering leadership

Software development

Team management

Agile methodologies

Product partnership

Planning and delivery

Tools

AWS Neuron SDK

Trainium

Inferentia

Job description

Annapurna Labs (U.S.) Inc. in Cupertino, CA is seeking a Sr. Software Development Manager to lead a team developing components and libraries enabling AI model optimization on Trainium and Inferentia accelerators within the AWS Neuron SDK.

You will collaborate with ML model developers and cross-functional teams to deliver high-performance, usable SDK capabilities and shape the strategic direction with senior leadership.