ML Accelerator Systems Software Engineer

Annapurna Labs; Cupertino (CA); On-site; USD 134,000 - 163,000; Full time

Job summary

Annapurna Labs (U.S.) Inc. in Cupertino, CA is seeking a software engineer to build and optimize ML accelerator software. You will work across hardware-software boundaries on high-performance systems and contribute to compiler tooling, drivers, and runtime libraries.

You will write C++, Python, and possibly Rust in a collaborative, fast-paced environment, with opportunities to influence silicon bring-up, ML inference workloads, and production-scale deployments at scale.

Qualifications

• Master's degree in CS/CE required.
• Experience programming with C++ using OO.
• Coursework in Computer Architecture.

Responsibilities

• Collaborate with cross-disciplinary teams to develop software for ML accelerator silicon.
• Design, develop, and test software across the accelerator stack.
• Build scalable systems for silicon bring-up and ML workloads.
• Write clean code in C++, Python, or Rust and develop CI/CD pipelines.

Skills

C++ (OO)

Python

Rust

Education

Master's degree in Computer Science or Computer Engineering

Tools

gem5

QEMU

self-built simulator

PyTest

GoogleTest

Job description

Annapurna Labs (U.S.) Inc. in Cupertino, CA is seeking a software engineer to build and optimize ML accelerator software. You will work across hardware-software boundaries on high-performance systems and contribute to compiler tooling, drivers, and runtime libraries.

You will write C++, Python, and possibly Rust in a collaborative, fast-paced environment, with opportunities to influence silicon bring-up, ML inference workloads, and production-scale deployments at scale.