Research Engineer, Learnable Planner (Integration) jobs in United States
cer-icon
Apply on Employer Site
company-logo

Waabi · 7 hours ago

Research Engineer, Learnable Planner (Integration)

Waabi is an AI company focused on developing the next generation of self-driving technology. The Research Engineer for Learnable Planner will integrate new AI technologies into the autonomy stack, contributing to the launch of fully driverless autonomous trucks and enhancing the planner architecture.

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

Responsibilities

Integrate cutting-edge ML models in production planning stack from development to validation, deployment, and monitoring
Develop necessary interfaces and pipelines in simulation for testing prototype or production planning models
Work closely with motion planning sub-teams and research scientists to improve our planner architecture and develop rich and novel representations that can facilitate end-to-end solutions
Champion engineering excellence, ensuring high-quality, well structured and tested code
Stay up-to-date with the latest advancements in the field of artificial intelligence, machine learning, computer vision, and self-driving technologies, and apply insights from the literature
Work with large datasets from various sources as well as Waabi World, our high-fidelity simulator
Contribute to the publication of research findings in conferences as well as Waabi's blog

Qualification

Machine LearningDeep LearningPythonC++PyTorchReinforcement LearningResearch PublicationCode EfficiencyCollaboration

Required

MS/PhD in machine learning, computer science, engineering, or a related field. Exceptional Bachelor's students will also be considered
Experience in ML-based or classical techniques for planning/decision making (e.g., imitation and reinforcement learning, optimization-based approaches, search methods, probabilistic reasoning)
Passion for taking research ideas and turning them into practical solutions for real-world applications
Open-minded and collaborative team player with willingness to help others
Solid understanding of computing fundamentals, including code efficiency
Experience in deep learning frameworks such as PyTorch
Proficiency in Python, Rust, C++ and/or CUDA

Preferred

Experience deploying ML/DL models to a production motion planning or related robotics stack
Experience in iterating on a model including evaluation, introspection and fine-tuning
Strong grasp of machine learning literature, including current trends and state-of-the-art techniques
Comfortable with model compilation and exporting, lower level concepts like TensorRT, CUDA kernels

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.

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
Company data provided by crunchbase