Amazon

AI Engineer - VLA model, RIVR

Amazon
Apply
15 hours 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.
  • Streamline data collection and model-training workflows for new tasks and data sources.
  • Collaborate with the reinforcement learning team using simulated and real-world data.
  • Optimize and distill neural networks for real-time deployment on edge hardware such as Nvidia Jetson Thor.
  • Build, lead, and mentor a software engineering team.
  • Advise product managers and executives on strategic decisions.
  • Create and maintain documentation, guidelines, and best practices.

Requirements

  • At least three years of industry or research experience, with applicable PhD experience accepted.
  • Strong fundamentals in supervised learning, 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 experience, 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.
  • Master's degree or higher in Engineering, Robotics, Machine Learning, or a related field.
  • Preferred: PhD in Robotics, Engineering, Computer Science, Machine Learning, or a similar discipline, or equivalent 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