Amazon

Software Development Engineer I, ML Infra Services, Annapurna Labs

Amazon$127K — $185K *
Information Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • 5-7 years experience in software development using modern programming languages like Java, Python, C++, or C#
  • Proficiency in object-oriented design and data structures
  • Familiarity with CI/CD pipelines and version control systems like Git
  • Experience in optimization mathematics, particularly in linear and nonlinear optimization
  • Knowledge of distributed systems and big data architectures

Responsibilities

  • Design and implement tooling for ML workloads on custom AI accelerators.
  • Build impactful solutions for a large customer base.
  • Engage in design discussions and code reviews with cross-functional teams.
  • Automate metrics collection and resolve software defects.
  • Work within a fast-paced, startup-like team environment.

Benefits

  • Comprehensive health insurance including medical, dental, and vision.
  • 401(k) matching and paid parental leave.
  • Flexible spending accounts and adoption reimbursement.
  • Mental health support services and resources.
  • Access to medical advice and counseling plans.
Full Job Description
We're looking for a Software Development Engineer to help build and evolve machine learning tools that run, optimize, and analyze ML workloads on custom AI accelerators. You'll work across the stack, from infrastructure orchestration to developer-facing tooling - alongside hardware engineers, system architects, and ML researchers both within and outside Amazon.

Key job responsibilities

- Design and implement tooling for profiling, optimization, and resource management of ML workloads on custom accelerators.

- Build high-impact solutions that ship to a large and growing customer base.

- Participate in design discussions, code reviews, and cross-functional collaboration with hardware, software, and customer-facing teams.

- Create metrics, implement automation, and resolve root causes of software defects.

- Work in a startup-like environment where you're always focused on the most important problems.

About the team

This is a high-impact, high-visibility team where your work directly accelerates every Neuron team's ability to ship, effectively multiplying the output of 100+ engineers. We're a small, senior group actively building greenfield capabilities, which means significant design ownership for SDEs and the opportunity to own major components and drive architectural decisions. You'll work at the cutting edge of AI infrastructure, at the intersection of Kubernetes, custom silicon, and large-scale ML workloads.

BASIC QUALIFICATIONS

- Are 18 years of age or older

- Experience with at least one modern language such as Java, Python, C++, or C# including object-oriented design

- Experience with at least one general-purpose programming language such as Java, Python, C++, C#, Go, Rust, or TypeScript

- Experience with data structure implementation, basic algorithm development, and/or object-oriented design principles

- Proficiency in Java and at least one of Go, Python, or TypeScript.

- Familiarity with Git and CI/CD pipelines.

PREFERRED QUALIFICATIONS

- Experience from a technical internship

- Experience in optimization mathematics such as linear programming and nonlinear optimization

- Experience with distributed, multi-tiered systems, algorithms, and relational databases

- Work 40 hours/week, and overtime as required

- Experience from previous technical internship(s) or demonstrated project experience

- Experience with Cloud platforms (preferably AWS), database systems (SQL and NoSQL), AI tools for development productivity, contributing to open-source projects, and/or version control systems

- Internship or project experience with AWS services (EKS, EC2, Lambda, S3, DynamoDB, or SQS).

- Familiarity with distributed systems or big data architectures.

- Experience with Linux systems and performance profiling.

- Exposure to compiler toolchains, code generation, or instruction set architectures (CPU, NPU, GPU).

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, CA, Cupertino - 127,100.00 - 185,000.00 USD annually

About Amazon

Audible is a provider of spoken audio information and entertainment , on the Internet. They provide premium spoken audio content, such as audio versions of books and newspapers and radio programs, that is delivered over the Internet and played back on personal computers and hand-held electronic devices. The Audible service allows consumers to purchase and download their content from their Website, store it in digital files and play it back on personal computers and electronic devices. More than 15,000 hours of audio content are available on their Web site, including audio versions of books, periodicals and radio programs. Several manufacturers have agreed to support and promote the playback of their content on their hand-held audio-enabled electronic devices.

Amazon Careers

Joining Amazon presents an unparalleled opportunity to become part of a vibrant team pushing the boundaries of innovation and growth in the global marketplace. As a leader in e-commerce, technology, and logistics, Amazon offers a variety of job opportunities that cater to a range of skills and professional interests. Work You’ll Do At Amazon, every day is an opportunity to collaborate with the brightest minds in technology and business to redefine what’s possible. Whether you’re interested in software development, marketing, human resources, or customer service, Amazon has a position waiting for you. Transform the way the world shops and innovates with our diverse and inclusive team. Amazon is not just a company; it’s a community where you can drive real change and contribute to projects impacting millions globally. Lead with Innovation and Leadership Amazon is the perfect place to enhance your leadership and innovation skills. Our culture encourages pushing the envelope and imagining the unimaginable. Here, you will lead projects that challenge the status quo and define new industry standards. Work with a team that values diversity and is committed to creating an inclusive environment. Our leadership is focused on harnessing the collective power of unique perspectives to foster growth and innovation. Explore Amazon’s Employment Benefits Amazon’s commitment to its employees extends beyond just career growth. We offer competitive benefits, including health care, parental leave, and diversity training, ensuring that our team not only excels professionally but also enjoys well-being and security. Internship and Networking Opportunities Start your career with an Amazon internship and gain hands-on experience that matters. Our internships provide a gateway to full-time employment and an opportunity to network with professionals across various sectors of the company. Future-Proof Your Career With Amazon, your career path is filled with numerous opportunities for advancement. Our learning and development programs are designed to nurture your professional growth and keep you at the forefront of industry trends. Stay Connected Join Our Team Discover the job opportunities at Amazon that match your skills and interests. We are constantly on the lookout for passionate, curious, and innovative team players ready to make a difference. Keep Up to Date Stay ahead with career tips, insider perspectives, and industry-leading insights you can put to use today—all from the people who work here. Job Alert Emails Customize your subscription to receive job alerts, the latest news, and insider tips tailored to your preferences. Explore the exciting and rewarding career opportunities that await at Amazon. Amazon is more than just a company—it’s a platform for building a promising future. Whether you’re starting or looking to advance your career, Amazon offers the resources, support, and network you need to succeed. Join us, and be a part of our continuing mission to be Earth's most customer-centric company.
Learn more about Amazon
Size
1,608 employees
Market Cap
$832.6 billion
Industry
Net Income
$21.3 billion
Founded
1994
5 Year Trend
+28.1%
Revenue
$386 billion
NASDAQ

Similar Jobs

More Jobs at Amazon

More Information Technology Jobs

Find similar Software Development Engineer I, ML Infra Services, Annapurna Labs jobs: