Thinking Machines LabRemote · San Francisco$350k-$475k4h ago
AmazonPosted 3w ago
Sr. Applied Scientist, SSG Science at Amazon scores 97 out of 100 on AI centrality, which makes it AI Level 4 of 4 (Builds AI) on this board. The level measures how much of the work is AI, not seniority.
IN, KA, Bengalurumidfull-time
AI in this role
Key job responsibilities
What will you do?
- Quantize, prune, distill, finetune Gen AI models to optimize for edge platforms
- Fundamentally understand Amazon’s underlying Neural Edge Engine to invent optimization techniques
- Analyze deep learning workloads and provide guidance to map them to Amazon’s Neural Edge Engine
- Use first principles of Information Theory, Scientific Computing, Deep Learning Theory, Non Equilibrium Thermodynamics
- Train custom Gen AI models that beat SOTA and paves path for developing production models
- Collaborate closely with compiler engineers, fellow Applied Scientists, Hardware Architects and product teams to build the best ML-centric solutions for our devices
- Publish in open source and present on Amazon's behalf at key ML conferences - NeurIPS, ICLR, MLSys.
Basic qualifications
- 3+ years of building machine learning models for business application experience
- PhD, or Master's degree and 4+ years of applied research experience
- Experience programming in Java, C++, Python or related language
- Experience with neural deep learning methods and machine learning
- Prior experience in productionizing ML models and managing their full lifecycle from development to deployment
Preferred qualifications
- Understanding and preferably hands-on experience with recent methods for inference optimization, including Mixture-of-Experts (MoE), Diffusion Models for Language Generation, etc.
Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.
Similar roles
Research roles rated AI Level 4 at other companies.
$350k-$475k4h ago
TuringPalo Alto, California, United States; San Francisco, California, United States; Seattle, Washington, United States$250k-$400k4h ago
$250k-$400k4h ago
PerplexityBerlin1d
pytorchdeepspeedfsdp
1d
NVIDIAUS, CA, Santa Clara$38-$94/hr1d
pytorchtensorflowjaxpythoncuda
$38-$94/hr1d
$199k-$325k2d
MercorRemote · San Francisco$5000k2d
$5000k2d
WaymoRemote · Mountain View, CA, USA; San Francisco, CA, USA; New York, NY, USA$213k-$263k2d
pytorchtensorflowjaxpython
$213k-$263k2d
2d
What kind of AI work fits you?
Answer 12 practical questions in about three minutes. Get a simple profile, the work it points to, and live roles to explore next.
Find my next step





