Thinking Machines LabRemote · San Francisco$350k-$475k4h ago
AmazonPosted 3w ago
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
- 2+ years of building machine learning models for business application experience
- PhD, or Master's degree and 2+ years of applied research experience
- Experience in patents or publications at top-tier peer-reviewed conferences or journals
- Experience programming in Java, C++, Python or related language
- Experience in any of the following areas: algorithms and data structures, parsing, numerical optimization, data mining, parallel and distributed computing, high-performance computing
- Strong foundation in relevant mathematical concepts (linear algebra, calculus, etc.)
Preferred qualifications
- Experience in professional software development
- Prior experience in productionizing ML models and managing their full lifecycle from development to deployment
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





