Jobs · Engineering · Washington

Machine Learning Engineer, AWS Neuron Inference, Annapurna ML

Amazon Web Services (AWS) · Seattle, WA · Today
EngineeringFull-time

About The Team

Our team is dedicated to supporting new members. We have a broad mix of experience levels and tenures, and we’re building an environment that celebrates knowledge-sharing and mentorship. Our senior members enjoy one-on-one mentoring and thorough, but kind, code reviews. We care about your career growth and strive to assign projects that help our team members develop your engineering expertise so you feel empowered to take on more complex tasks in the future.

Key Job Responsibilities

  • Develop optimized building blocks for the Neuron distributed inference library, tuning them to ensure highest performance and maximize efficiency running on Trn2 and Trn3 servers.
  • Create metrics, implement automation and other improvements, and resolve the root cause of software defects as you develop technology components.
  • Participate in design discussions and code review.
  • Communicate with internal and external stakeholders.
  • Work cross-functionally with teams across Neuron in a fast-paced startup-like development environment, constantly staying on top of the latest priorities as the AI landscape evolves.

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

Pay & Benefits

  • Base salary range: $143,700.00 - $194,400.00 USD annually (USA, WA, Seattle)
  • Sign-on payments and restricted stock units (RSUs)
  • 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
  • Parental leave
This response is AI-generated, for reference only.

Similar jobs