Member Technical Staff - Reinforcement Learning jobs at Amazon in United States, San Francisco

Key job responsibilities
- Design and implement highly scalable and realistic simulation environments for training agents using reinforcement learning.
- Develop and deploy generative AI approaches to automate the creation of diverse and complex simulation environments and scenarios.
- Analyze, troubleshoot, and profile complex machine learning systems within the simulation context, identifying and resolving performance bottlenecks.

- PhD, or Master's degree or 5+ years of applied research experience
- Experience programming in C++, Python or related language
- Experience with simulation and synthetic data generation
- Experience with neural deep learning methods and machine learning

- PhD in Computer Science, Machine Learning, or a related field
- 3+ years' experience building machine learning models (includes internships)
- Demonstrated experience in developing and implementing simulation environments or synthetic data generation for reinforcement learning.
- Strong programming skills in Python and experience with deep learning frameworks such as Tensor Flow or PyTorch
- Excellent problem-solving skills, with the ability to think creatively and critically about complex problems
- Strong communication and collaboration skills, with the ability to work effectively with cross-functional teamsPursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

Key job responsibilities
You will contribute directly to AI agent development in an applied research role, including model training, dataset design, and pre- and post-training optimization. You will be hired as a Member of Technical Staff.

- 3+ years' experience building machine learning models
- Proficiency in Python, Java, C++, or related language
- Experience with deep learning methods and tools, e.g., PyTorch, JAX

- PhD or Master's degree in computer science or related field
- Strong background in scientific research with a proven ability to generate and implement new ideas in machine learning
- Willingness to step outside typical role boundaries to get things done — every member of technical staff is expected to write code, design experiments, and interpret results
- Ability to communicate results and insights to both technical and non-technical audiences, including through presentations and written reports
- Ability to think big about the arc of development of AI over a multi-year horizon, and identify new opportunities to apply these technologies to solve real-world problems
- Capacity to mentor and guide junior scientists and engineers, and contribute to the overall growth and development of the teamPursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

Key job responsibilities
* Design, build, and deploy machine learning models, frameworks, and data pipelines
* Optimize ML training, inference, and evaluation workflows for reliability and performance
* Evaluate and improve ML model performance and metrics
* Develop tools and infrastructure to enhance ML development productivity

* 3+ years experience building and deploying machine learning systems
* Experience with ML frameworks (e.g. PyTorch, TensorFlow) and ML orchestration tools
* Proven track record of improving ML systems, performance, and workflows

* Experience building and optimizing production-grade AI systems, e.g., for self-driving, simulation, or LLM applications
* Willingness to step outside typical role boundaries to get things done — every member of technical staff is expected to write code, design experiments, and interpret results
* Capacity to work autonomously and with a small team to drive research progressPursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

Key job responsibilitiesThe ideal candidate is passionate about new opportunities and has a demonstrable track record of success in delivering new features and products. A commitment to team work, hustle, and strong communication skills (to both business and technical partners) are absolute requirements. Creating reliable, scalable, and high performance AI products requires exceptional technical expertise, a sound understanding of the fundamentals of Computer Science and Machine Learning. This person has thrived and succeeded in delivering high quality technology products/services in a hyper-growth environment.

- 3+ years of non-internship professional software development experience
- 2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience
- Experience programming with at least one software programming language

- 3+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
- Bachelor's degree in computer science or equivalent

In this role, you will work closely with research teams to design, build, and maintain systems for training and evaluating state-of-the-art agent models.Key job responsibilities
* Develop cutting-edge training infrastructure to ensure large-scale reinforcement learning on LLMs runs highly efficient and robust.
* Work across the entire technology stack, including low level ML system, job orchestration and data management.
* Analyze, troubleshoot and profiling complex ML systems, identify and address performance bottlenecks.
* Work closely with researchers, conduct MLSys research to create new techniques, infrastructure, and tooling around emerging research capabilities.

- PhD, or Master's degree and 3+ years of applied research experience
- Experience programming in Java, C++, Python or related language
- Experience with neural deep learning methods and machine learning
- Experience debugging ML systems

- PhD in Computer Science, Machine Learning, or a related field, with a focus on ML System.
- Demonstrated experience in developing, implementing and debugging large scale ML systems.
- Experience with distributed system, Megatron, vLLM, Ray, and working with GPUs.
- Experience with patents or publications at top-tier peer-reviewed conferences or journals.Pursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

Key job responsibilities
You will contribute directly to AI agent development in an applied research role, including model training, dataset design, and pre- and post-training optimization. You will be hired as a Member of Technical Staff.

- 3+ years' experience building machine learning models
- Proficiency in Python, Java, C++, or related language
- Experience with deep learning methods and tools, e.g., PyTorch, JAX

- PhD or Master's degree in computer science or related field
- Strong background in scientific research with a proven ability to generate and implement new ideas in machine learning
- Willingness to step outside typical role boundaries to get things done — every member of technical staff is expected to write code, design experiments, and interpret results
- Ability to communicate results and insights to both technical and non-technical audiences, including through presentations and written reports
- Ability to think big about the arc of development of AI over a multi-year horizon, and identify new opportunities to apply these technologies to solve real-world problems
- Capacity to mentor and guide junior scientists and engineers, and contribute to the overall growth and development of the teamPursuant to the San Francisco Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.

Export Control Requirement: Due to applicable export control laws and regulations, candidates must be a U.S. citizen or national, U.S. permanent resident (i.e., current Green Card holder), or lawfully admitted into the U.S. as a refugee or granted asylum.
Key job responsibilities
We are looking for an experienced engineer to lead design verification and test for the Kuiper satellite electrical power system. In this role you will be responsible for ensuring the satellite electrical power system meets all requirements through testing in different venues and levels of integration. This will include:• Working with system engineers to create verification cross reference matrices to define test coverage and execution across all power system test venues including hardware in the loop and production acceptance testing.
• Using power system design requirements to develop test plans and procedures.
• Define and develop test systems/venues necessary to meet verification goals of the power system, including development and test execution on hardware in the loop integrated test beds.
• Drive test execution for system level verification testing, and support teams working on production test execution.
• Drive test readiness reviews, and verification/qualification test results reviews with multiple cross-team leaders
• Be a technical leader/role model and mentor of other engineers; instilling best practices across the full development and test lifecycles.A day in the life
In this role you will be a member of the satellite power systems team which oversees development and validation of the current and future generation satellite electrical power systems. As a Sr. Power Systems Test Engineer, you will own power system verification and test and will work with other power systems and development engineers to deliver a fully validated power system for the satellite. You will work to develop test plans for the satellite power system that provides full coverage for system requirements. You will function as a lead for power hardware-in-the-loop (PHITL) test beds, ensuring these venues are properly designed to provide verification of the satellite power system. You will also drive reviews for a given campaign of power system validation; pulling in cross-functional teams to review and reach alignment. You will also support production test venues through test definition and data reviews. Finally, as a senior engineer on the team, you will serve as a technical leader and role model, ensuring that the team upholds a high standard, and instilling best practices within the power systems engineering team.
About the team
You will work with a team of top-notch engineers and collaborate with multiple other spacecraft design and integration teams. You should be comfortable with a high degree of ambiguity and relish the idea of solving problems that haven't been solved at scale before. You’ll have relentlessly high standards for yourself and everyone you work with, and you’ll be constantly looking for ways to improve system performance, and manage non-conformances into actionable items and procedural improvements. We want individuals who are ready for this challenge and want to reach beyond what is possible today.

- Experience in developing functional specifications, design verification plans and functional test procedures
- BS degree in electrical engineering or equivalent.
- 5+ years of experience in design or systems architecture role working within power systems or related area of development.
- Experience with complex power systems design.

- Master's degree in electrical engineering, computer engineering, or equivalent
- Experience leading interdisciplinary teams to execute product design from concept to production
- Experience with development of hardware in loop testing for automotive and/or aerospace systems.
- System test verification experience in the area of power conversion, power systems, solar, battery, electric vehicles, satellites, or rockets.
- Experience in requirements definition and derivation using modelling, simulation and other analysis tools.
- Experience as a technical lead or manager.
- Experience developing system architecture for multiple subsystems.
- Experience with low-level software development and/or scripting with applications involving testing of power components or integrated power systems.
- Working knowledge of at least one programming language such as Python or C/C++.
- Cross-organizational communication – comfortable evangelizing ideas and concerns clearly and persuasively, in both verbal and written forms