About the Role

This senior software engineering role focuses on developing power and performance management software for next-generation ML acceleration chips at Annapurna Labs. The engineer will implement control algorithms, build tracing tools, and create automated workflows to optimize machine learning workloads.

Key Responsibilities

• Develop and implement software/firmware for managing power, thermal, and performance behavior
• Translate architectural specifications and models into working, tested code
• Participate in design, implementation, and validation of control algorithms across chip, card, and server scope
• Collaborate with power architects and system teams for software-hardware integration
• Develop and automate tools/frameworks for running training/inference workloads and collecting metrics

Required Skills & Qualifications

Must Have:

• 5+ years of non-internship professional software development experience
• Experience as a mentor, tech lead or leading an engineering team
• Bachelor's or Master's degree in Computer Science, Electrical/Computer Engineering, or related field
• Proficiency in C/C++ and scripting languages (e.g., Python)

Nice to Have:

• Knowledge of machine learning frameworks (e.g., TensorFlow, PyTorch) and related workloads
• Exposure to system software development (kernel drivers, firmware, embedded systems)
• Familiarity with hardware/software interactions or hardware-aware programming
• Experience with thermal, power, or performance management

Benefits & Perks

• Mentorship and career growth opportunities
• Work-life balance and flexible working culture
• Inclusive team culture with employee-led affinity groups
• Competitive compensation with geographic market adjustments
• Comprehensive medical, financial, and other benefits