Back to jobsJob overview

About the role

Software Development Engineer, Annapurna Labs, Machine Learning Acceleration - Power and Performance at Annapurna Labs (U.S.) Inc.

Required Skills

pythonc++machine learningembedded systemspower managementperformance optimizationfirmwaredebugging

About the Role

Software Development Engineer role focusing on power and performance management for next-generation ML acceleration chips. Responsibilities include developing software algorithms for power/thermal/performance control and building tools to optimize training/inference workloads. This position requires strong software engineering skills with opportunities to work across the software-hardware stack.

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 workloads and collecting power/performance metrics

Required Skills & Qualifications

Must Have:

  • Bachelor's degree in computer science, electrical engineering, or related field
  • 3+ years software development skills across full software development lifecycle
  • Proficiency in C/C++ and scripting languages (e.g., Python)
  • Experience writing production-quality code and working in collaborative, cross-functional teams
  • Strong debugging and problem-solving skills

Nice to Have:

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

Benefits & Perks

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