Thinking Machines LabRemote · San Francisco$350k-$475k6h ago
AmazonPosted 1w ago
Applied Scientist, One MHS - Software, Controls, Science at Amazon scores 99 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.
AI in this role
Key job responsibilities
• Own the research and development of optimization and sequential decision-making solutions spanning constraint programming, stochastic and robust optimization, contextual bandits, and reinforcement learning for real-time MHE control and scheduling optimization in a production environment.
• Formulate fulfillment operations and manufacturing scheduling problems (production scheduling, resource allocation, sorter optimization, throughput and congestion control) as optimization or sequential decision-making problems, and design multi-objective functions that balance competing operational objectives such as on-time delivery, utilization, changeover cost, and schedule stability.
• Build and leverage high-fidelity simulation and emulation environments for safe offline training, policy validation, and transfer to live systems before fleet-scale deployment.
• Collaborate across multiple science and engineering teams to integrate policies into production planning and real-time control systems, including monitoring, guardrails, and staged rollout.
• Communicate results and their limitations clearly in writing to technical and business audiences, and contribute to the team's external research presence through publication where the work merits it.
About the team
Amazon is building next generation software, hardware, and processes that will run our global network of fulfillment centers that move millions of units of inventory, and ensure customers get what they want when promised.
The Science Software team in the One MHS organization unlocks Material Handling Equipment (MHE) innovation through a multiplicity of disciplines within Artificial Intelligence (AI) and applied science, including Optimization, Reinforcement Learning, classical Machine Learning, statistical modeling, Computer Vision (CV), and Physics-Informed Neural Networks (PINNs). The team is dedicated to building self-optimizing fulfillment centers, developing the models that drive real-time, building-wide orchestration of MHE. We conduct experiments, develop models, and apply machine learning (ML) at scale to optimize throughput, flow, merge, and congestion control, and to improve operational performance across the fulfillment network.
Basic qualifications
- 2+ years of building machine learning models or developing algorithms for business application experience
- PhD in Operations Research, Statistics, Applied Mathematics, Engineering, Computer Science or related field
- Experience in optimization mathematics such as linear programming and nonlinear optimization
- Knowledge of and proficiency in the use of Python scripting language
- Experience Experienced with end-to-end ownership of major project deliverables
- Experience with popular deep learning frameworks and RL tooling (e.g., PyTorch, d3rlpy, Ray/RLlib, Gymnasium, Stable-Baselines3, Isaac Gym/Omniverse)
- Demonstrated experience developing and applying optimization or reinforcement learning solutions (e.g., MILP, constraint programming, stochastic programming, contextual bandits, deep RL) to real-world control, scheduling, or operation problems
Preferred qualifications
- First-author publications at top-tier machine learning, operations research, or control venues (e.g., NeurIPS, ICML, ICLR, AAAI, AISTATS, CPAIOR, INFORMS Journal on Computing, or IEEE control and automation conferences)
- Experience building a discrete-event simulator to train and evaluate operational policies, and calibrating it against historical data
- Experience applying optimization or RL in a setting analogous to ours: production scheduling, real-time industrial control, robotics, material handling, industrial process or operations.
- Experience deploying optimization or ML models to production at scale and partnering with engineering teams on inference, monitoring, and feedback loops
Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.
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.
The base salary range for this position is listed below. Your Amazon package will include sign-on payments and restricted stock units (RSUs). Final compensation will be determined based on factors including experience, qualifications, and location. Amazon also offers comprehensive benefits including health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage), 401(k) matching, paid time off, and parental leave. Learn more about our benefits at https://amazon.jobs/en/benefits.
USA, MA, Boston - 142,800.00 - 193,200.00 USD annually
USA, MA, North Reading - 142,800.00 - 193,200.00 USD annually
Prepare for this job
A free preview built only from this posting: what it asks for, what you could be asked in an interview, and how to adjust your resume.
Skills and AI tools this role asks for
Questions you could be asked
- Walk me through a computer vision problem you solved, from raw data to a deployed model.
- Walk me through how you've used PyTorch in your day-to-day work.
- How would you decide a model or AI system is ready to ship?
- Tell me about a time a model underperformed in production. How did you find out, and what did you change?
Adapt your resume
- List these exact terms on your resume: Computer Vision and PyTorch. An applicant tracking system matches the wording, not the idea.
- Attach one line of real, concrete experience to at least one of them — a tool named with nothing behind it rarely survives a human read.
- Lead with what you built, trained or shipped — this role is judged on the AI system itself, not the tools around it.
Want your resume actually rewritten for this job?
The free preview above is everything we have today. A full resume rewrite is not live yet and has no price set. Join the waitlist and we will email you if we open it.
Similar roles
Research roles rated AI Level 4 at other companies.
TuringPalo Alto, California, United States; San Francisco, California, United States; Seattle, Washington, United States$250k-$400k6h ago
PerplexityBerlin1d
NVIDIAUS, CA, Santa Clara$38-$94/hr1d
MercorRemote · San Francisco$5000k2d
WaymoRemote · Mountain View, CA, USA; San Francisco, CA, USA; New York, NY, USA$213k-$263k2d
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





