Senior / Staff Research Engineer, Simulation Agents jobs in United States
cer-icon
Apply on Employer Site
company-logo

Waabi · 4 hours ago

Senior / Staff Research Engineer, Simulation Agents

positionCanada, KY
timeFull-time
remoteHybrid
senioritySenior Level, Lead/Staff
money$158K/yr - $269K/yr

Waabi is an AI company building the next generation of self-driving technology. As a Senior/Staff Research Engineer on the Behaviors team, you will create simulation software for autonomous driving and work closely with a multidisciplinary team to develop scalable solutions.

Artificial Intelligence (AI)AutomotiveAutonomous VehiclesMachine Learning
check
H1B Sponsor Likelynote

Responsibilities

Architect robust and scalable training and evaluation frameworks for IL/RL-based traffic agents
Design and implement scalable data pipelines for generating large-scale training datasets
Productionize cutting-edge research into robust, scalable systems in Waabi World, our simulation platform
Contribute to technical direction and strategy for realistic traffic agents and scenario generation

Qualification

Imitation learningReinforcement learningML frameworksSoftware engineeringScalable ML systemsTraffic simulationWillingness to helpPublications in conferencesRust/C++Open-mindedCollaborative team player

Required

Experience with imitation learning, reinforcement learning, planning, and/or robotics
Expertise with ML frameworks such as PyTorch, TensorFlow, or Jax
Proven ability to take research from prototype to production systems
Excellent software engineering skills with experience building scalable ML systems
Open-minded and collaborative team player with willingness to help others

Preferred

Masters/PhD in machine learning, computer science, engineering, or a related field
Publications in top-tier conferences or journals related to machine learning or robotics
Experience in self-driving, traffic simulation, or a related field
Expertise coding in Rust/C++

Benefits

Competitive compensation and equity awards.
Health and Wellness benefits encompassing Medical, Dental and Vision coverage (for full-time employees only).
Unlimited Vacation.
Flexible hours and Work from Home support.
Daily drinks, snacks and catered meals (when in office).
Regularly scheduled team building activities and social events both on-site, off-site & virtually.

Company

Waabi

twittertwittertwitter
company-logo
Waabi is pioneering Physical AI starting with autonomous trucking.
dateFounded in 2021
location
Toronto, Ontario, CAN
people-size51-200 employees
web-link
https://waabi.ai

H1B Sponsorship

Waabi has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (8)
2024 (6)
2023 (6)
2022 (4)

Funding

Current Stage
Growth Stage
Total Funding
$283.04M
Key Investors
Volvo Group Venture CapitalKhosla Ventures
2024-06-18Series B· $200.48M
2023-01-18Series Unknown
2021-06-08Series A· $82.55M

Leadership Team

leader-logo
Raquel Urtasun
Founder & CEO
linkedin
leader-logo
Lior Ron
COO
linkedin

Recent News

Calgary Herald
Canada has problems: Here are the tech startups worth watching that are trying to tackle them
2026-01-08
Globe and Mail
Lately: Australia bans social media for kids, Mickey Mouse is coming to Sora, and a must-watch sci-fi thriller
2025-12-13
The Star
For Canada’s startups, AI was a dominant force in 2025
2025-12-13
Company data provided by crunchbase