SDE, MLA hardware/software co-design, Annapurna Labs Machine Learning Acceleration

Amazon.com; Austin (TX); On-site; USD 144,000 - 194,000; Full time

Benefits offered by this job

RSUs

Health insurance

401(k) matching

Paid time off

Job summary

Amazon is seeking an SDE in the pre-silicon team to drive HW/SW co-development for Trainium ML chips. You will work with architecture, design and emulation teams to build new silicon functionality and write bare-metal software to verify end-to-end SoC performance.

You should have 3+ years of professional software development, 2+ years in design/architecture, and experience in at least one programming language. Amazon offers an inclusive environment and comprehensive benefits.

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.
• Experience programming with at least one software programming language.

Responsibilities

• Drive pre-silicon hardware/software co-development for Trainium machine learning chips.
• Collaborate with architecture, design and emulation teams to build silicon functionality.
• Write bare-metal software to verify the end-to-end functionality of the SoC and subsystems.
• Ensure reliability and performance of various subsystems in the SoC.

Skills

Software development

System design/architecture

Programming languages

Education

Bachelor's degree in Computer Science or equivalent

Job description

AWS Utility Computing (UC) provides product innovations — from foundational services such as Amazon’s Simple Storage Service (S3) and Amazon Elastic Compute Cloud (EC2), to consistently released new product innovations that continue to set AWS’s services and features apart in the industry. As a member of the UC organization, you’ll support the development and management of Compute, Database, Storage, Internet of Things (IoT), Platform, and Productivity Apps services in AWS, including support for customers who require specialized security solutions for their cloud services.
Custom machine learning chips are at the heart of our Trainium machine learning instances, and we invite you to build them with us!
As an SDE in the pre-silicon team, you will be responsible for driving the pre-silicon hardware/software co-development for our machine learning chips.
You will work with architecture, design and emulation teams to build new silicon functionality.
You will write bare-metal software to verify the end-to-end functionality of the SoC and the functionality and performance of different subsystems in the SoC.
If you have a solid background in embedded software, system development or chip design, come join us!

Basic 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
• Experience programming with at least one software programming language

Preferred Qualifications

• 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
• Bachelor's degree in computer science or equivalent

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.

Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.

The base salary range for this position is listed below. Your Amazon package will include sign-on payments and restricted stock units (RSUs). Final compensation will be determined based on factors including experience, qualifications, and location. Amazon also offers comprehensive benefits including health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage), 401(k) matching, paid time off, and parental leave. Learn more about our benefits at https://amazon.jobs/en/benefits.

USA, TX, Austin - 143,700.00 - 194,400.00 USD annually