Annapurna Labs

Technology

SoftwareEngineerMLAcceleration

$70–194k Austin, Texas, United States FULL TIME
Market Sentiment
HIGH DEMAND

Neural analysis suggests this role is
optimal for Mid+ candidates.

The Brief

“Software Engineer ML Acceleration at Annapurna Labs. Skills: ML Acceleration, Distributed Systems, Hardware/Software Co-design. Design and implement ML acceleration techniques. Develop and optimize ML frameworks”

Industry & Context.

Technology
Problems you'll solve

Root cause analysis; Performance analysis

What They're Looking For.

Must Have

Bachelor's degree in Computer Science, 5+ years of experience in software development, Experience with large-scale distributed systems, Experience with machine learning acceleration techniques, Experience with hardware/software co-design

Nice to Have

Master's degree or PhD in a related field, Experience with ML frameworks like PyTorch or TensorFlow, Experience with GPU programming, Experience with compiler development, Experience with performance analysis tools

What You'll Do.

Design and implement ML acceleration techniques

Develop and optimize ML frameworks

Work on hardware/software co-design for ML

Analyze and improve performance of ML systems

Collaborate with researchers and engineers

Contribute to the development of new ML technologies

How You'll Work.

Team & Collaboration

Cross-functional teams; Researchers and engineers

Full Job Description

In Annapurna Labs we are at the forefront of hardware/software co-design not just in Amazon Web Services (AWS) but across the industry. The Machine Learning Server Software Team is looking for candidates interested in writing data-driven software for our Machine Learning servers. We help to maintain and manufacture the most advanced servers in the world. We touch technologies from the accelerator inner-workings to the I2C infrastructure pervasive in the server and everything in between. Our team does not work on machine learning algorithms, but rather on the physical systems (hardware) which execute and accelerate those machine learning algorithms. Data paths, I2C, accelerator inner-workings are our bread and butter. Come join our team. Key job responsibilities - Member of a team responsible for the software associated with server components and integration in to EC2. - Working with the MLA Hardware, Test and Manufacturing teams to create a coordinated software package to enable both qualification as well as rapid deployment of software. - Developing software (C/C++, Python, Lua) which can be maintained, improved upon, documented, tested, and reused. A day in the life The MLA Systems Software team was formed to focus on server software primarily for debug, testing, qualification, and manufacturing. At a high-level our goal is to find ways to help the organization scale though the use of software and automation. 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. Diverse Experiences AWS values diverse experiences. Even if you do not meet all of the

Free ATS check

Applying for this Software Engineer ML Acceleration role?

Most applicants get filtered before a human reads their resume. See if yours makes the cut.

ANONYMOUS · UNFILTERED

What do employees actually say about Annapurna Labs?

Real rants from real employees. Read before you apply.

Read Company Rants →