Sr. Firmware Engineer, Annapurna Labs, Machine Learning Acceleration - Power and Performance

Sr. Firmware Engineer, Annapurna Labs, Machine Learning Acceleration - Power and Performance

Join to apply for the Sr. Firmware Engineer, Annapurna Labs, Machine Learning Acceleration - Power and Performance role at Amazon Web Services (AWS)

AWS Utility Computing (UC) provides product Annapurna Labs, an AWS-owned organization that designs silicon and software accelerating innovation. Our custom chips, accelerators, and software stacks tackle technical challenges never seen before, enabling customers to solve problems that were unimaginable a short time ago. Join us to develop firmware that brings these innovations to life.

• Design and implement firmware algorithms for power management, thermal control, and performance optimization on ML acceleration hardware
• Develop real‑time control policies and state machines that dynamically balance power, thermal, and performance constraints
• Create optimization algorithms for resource allocation, frequency/voltage scaling, and workload scheduling
• Implement efficient data structures and algorithms suitable for embedded, resource‑constrained environments
• Design and implement on‑device tracing and telemetry collection systems to support algorithm development and validation
• Build developer tools and data pipelines for metric collection, analysis, and visualization of algorithm behavior
• Implement low‑overhead instrumentation that minimizes impact on workload performance
• Collaborate with hardware architects to understand hardware capabilities and identify optimal instrumentation points
• Develop automated testing and validation workflows; integrate with optional cloud‑based analytics pipelines
• Own firmware code quality through rigorous testing, debugging, and validation on hardware

You will work closely with power architects and hardware teams to understand silicon capabilities, implement low‑level control mechanisms, and create the algorithms and tooling that deliver optimal system behavior.

Our team supports new members, values knowledge sharing, and provides mentorship. Senior members conduct code reviews and help junior engineers grow. We encourage career development and assign projects that help you gain deeper expertise.

AWS values diverse experiences. Even if you do not meet all qualifications, we encourage you to apply. If your career is just starting or follows an alternative path, do not let that stop you.

Amazon Web Services is the world’s most comprehensive and broadly adopted cloud platform, trusted by startups to Global 500 companies.

Our affinity groups foster a culture of inclusion and diversity. We host events such as CORE and Amaze Con, celebrating our uniqueness.

We value flexibility and work‑life harmony, supporting both home and work goals.

We raise the bar through knowledge sharing, mentorship, and career‑advancing resources.

• 5+ years of non‑internship professional software development experience
• Experience as a mentor, tech lead, or leading an engineering team
• Bachelor’s degree in computer science, electrical engineering, or related field
• Strong firmware or embedded systems development experience
• Proficiency in C/C++ for systems programming with strong foundation in algorithms and data structures
• Experience implementing efficient algorithms in resource‑constrained, real‑time environments
• Experience with hardware interfaces, instrumentation, or performance monitoring
• Strong debugging skills with hardware‑software systems
• Experience building developer tools or instrumentation frameworks

• Experience developing control algorithms, optimization algorithms, or state machines in firmware
• Experience with power management algorithms, thermal control policies, or dynamic performance optimization
• Background in tracing frameworks, telemetry systems, or performance analysis
• Understanding of algorithmic complexity and optimization techniques for embedded systems
• Familiarity with hardware performance counters, on‑chip monitoring, or hardware debug interfaces
• Experience with data…