Senior / Staff Research Engineer, Simulation Agents
Company: Waabi
Location: Phoenix
Posted on: February 14, 2026
|
|
|
Job Description:
Job Description Job Description Waabi, founded by AI visionary
Raquel Urtasun, is the leader in Physical AI. With a world-class
team, we're unlocking the next era of autonomous transportation
with technology that's powering commercial autonomous trucks and
robotaxis. Waabi is backed by and partners with world leaders in
AI, automotive, logistics, and deep tech. With offices in Toronto,
San Francisco, Dallas, and Pittsburgh, Waabi is growing quickly and
looking for diverse, innovative and collaborative candidates who
want to impact the world in a positive way. To learn more visit:
www.waabi.ai The Behaviors team at Waabi develops cutting-edge
simulation agents and scenario generation algorithms for Waabi
World, our simulation platform. As a Senior/Staff Research Engineer
on our Behaviors team, you will create the next generation of
simulation software for autonomous driving. You will work closely
with our multidisciplinary team of research scientists and software
engineers to build innovative, practical, and scalable solutions to
challenging problems as part of our next-generation self-driving
solution. You will - 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. Qualifications: -
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.
Bonus/nice to have: - 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++. The US yearly salary
range for this role is: $158,000 - $269,000 USD in addition to
competitive perks & benefits. Waabi US Inc.’s yearly salary ranges
are determined based on several factors in accordance with the
Company’s compensation practices. The salary base range is
reflective of the minimum and maximum target for new hire salaries
for the position across all US locations. Note: The Company
provides additional compensation for employees in this role,
including equity incentive awards and an annual performance bonus.
Perks/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. - As we grow, this list continues to evolve!
Waabi is a technology start-up building technologies to transform
the way the world moves. Join our talented team to be a part of the
future and to make an impact! Waabi is an equal opportunity
employer. We celebrate diversity and are committed to creating a
supportive, inclusive, and accessible workplace for all our
employees. We seek applicants of all backgrounds and identities,
across race, color, ethnicity, national origin or ancestry, age,
citizenship, religion, sex, sexual orientation, gender identity or
expression, military or veteran status, marital status, pregnancy
or parental status, caregiver status, disability, or any other
characteristic protected by law. We make workplace accommodations
for qualified individuals with disabilities as required by
applicable law. If reasonable accommodation is needed to
participate in the job application or interview process please let
our recruiting team know. We may use artificial intelligence (AI)
tools to support parts of the hiring process, such as reviewing
applications, analyzing resumes, or assessing responses. These
tools assist our recruitment team but do not replace human
judgment. Final hiring decisions are ultimately made by humans. If
you would like more information about how your data is processed,
please contact us.
Keywords: Waabi, Peoria , Senior / Staff Research Engineer, Simulation Agents, Engineering , Phoenix, Arizona
