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 visualizationml hardwarepower optimizationdata analysisaws

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 in training and inference at massive scale.

Key Responsibilities

  • Analyze power consumption and performance metrics for ML acceleration workloads across data centers
  • Develop predictive models and statistical frameworks to identify optimization opportunities
  • Provide data-driven recommendations for server deployments and capacity planning decisions
  • 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 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, financial, and other benefits
  • Equity and sign-on payments may be provided
  • Workplace accommodations for disabilities