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