Back to jobsJob overview

About the role

Sr Data Scientist , Annapurna Labs, ML Acceleration Power Architecture at Annapurna Labs (U.S.) Inc. - D63

Required Skills

pythonsqlstatistical modelingdata visualizationmachine learningdata analysisawspower optimization

About the Role

Senior Data Scientist role focusing on power and performance optimization for ML acceleration workloads across Amazon's global data centers. Combines advanced data science with ML hardware expertise to drive efficiency improvements at massive scale. Involves predictive modeling, strategic planning, and cross-functional collaboration with engineering teams.

Key Responsibilities

  • Analyze power consumption and performance metrics for ML acceleration workloads
  • Develop predictive models and statistical frameworks to identify optimization opportunities
  • Provide data-driven recommendations for server deployments and capacity planning
  • Partner with software and hardware engineering teams to optimize ML frameworks and chip design
  • Conduct applied research on emerging ML acceleration technologies and energy efficiency

Required Skills & Qualifications

Must Have:

  • 5+ years of experience with data querying languages (SQL), scripting languages (Python) or statistical/mathematical software
  • 4+ years of data scientist experience
  • Experience with statistical models (e.g., multinomial logistic regression)

Nice to Have:

  • 2+ years of data visualization experience using AWS QuickSight, Tableau, R Shiny, etc.
  • Experience managing data pipelines
  • Experience as a leader and mentor on a data science team

Benefits & Perks

  • Medical benefits
  • Financial benefits
  • Other employee benefits
  • Equity compensation
  • Sign-on payments