Amazon

AI Engineer - VLA Foundation Model, RIVR

Amazon
Apply
1 day ago
Zürich, SwitzerlandMid Level

Responsibilities

  • Develop Vision-Language-Action models, generalist robot Transformers, and imitation-learning algorithms such as diffusion policies.
  • Design, test, and refine algorithms for real-world autonomy, navigation, spatial reasoning, and generalization.
  • Improve data-collection and training workflows to expand model capabilities across new tasks and data sources.
  • Collaborate with reinforcement-learning teams using simulated and real-world data.
  • Optimize and distill neural networks for real-time edge deployment, including on Nvidia Jetson Thor.
  • Build, lead, and mentor a team of software engineers.
  • Provide technical guidance to product managers and executives for strategic decisions.
  • Create and maintain documentation, guidelines, and best practices.

Requirements

  • Master’s degree or higher in Engineering, Robotics, Machine Learning, or a related field.
  • At least three years of industry or research experience, with applicable PhD experience accepted.
  • Strong fundamentals in supervised and self-supervised learning, Transformer architectures, policy optimization, imitation learning, generative AI, and diffusion models.
  • Proven experience developing Vision-Language-Action models or large-scale generalist robot models such as RT-2 or Octo.
  • Strong robotics background, including autonomy and navigation.
  • Experience deploying artificial neural networks on hardware platforms.
  • Ability to prototype algorithms and train deep neural networks using Python and PyTorch.
  • Preferred: PhD in Robotics, Engineering, Computer Science, Machine Learning, or a similar discipline, or an equivalent amount of research experience.
  • Preferred: publications at top-tier conferences.
  • Preferred: experience managing a software team.
  • Preferred: ability to write production-level code in modern C++.

Categories

Amazon

About Amazon

10,000+ employees

Amazon builds and operates a global e-commerce marketplace, logistics network, and consumer devices, and provides cloud computing via AWS for businesses and developers. The company earns revenue from online retail, third‑party seller services, subscriptions like Prime, advertising, and AWS usage. Founded in 1994 and headquartered in Seattle, it is publicly traded on NASDAQ (AMZN) and serves customers in dozens of countries.

Contact me