Research Engineer - Learnable Planner

Job not on LinkedIn

🕒 June 19, 2025

🏢🏡 San Francisco – Hybrid

💵 $159k - $296k / year

⏰ Full Time

🟡 Mid-level

🟠 Senior

📚 Research Engineer

🦅 H1B Visa Sponsor

This company has sponsored H1B visas in the past.

Apply Now

Find Similar Remote Jobs

Receive Emails For Remote Jobs

📊 Check your resume score for this job

Improve your chances of getting an interview by checking your resume score before you apply.

Upload Resume

Waabi

WebsiteLinkedInAll Job Openings

51 - 200 employees

Founded 2021

🚗 Transport

🤖 Artificial Intelligence

🔧 Hardware

💰 Venture Round on 2023-01

Transport • Artificial Intelligence • Hardware

Waabi is a company at the forefront of autonomous trucking technology. Led by AI and self-driving pioneer Raquel Urtasun, Waabi uses advanced Generative AI to develop scalable and safe self-driving solutions. The Waabi Driver, their cutting-edge product, blends an end-to-end trained AI stack with an interpretable architecture, facilitating OEM integration and large-scale commercialization. Waabi World, their highly advanced simulation engine, offers a realistic and immersive environment that accelerates the development and deployment of self-driving technology by reducing the reliance on extensive real-world testing. With partnerships and a visionary approach, Waabi aims to revolutionize logistics and supply chain industries.

📋 Description

• Waabi, founded by AI pioneer and visionary Raquel Urtasun, is an AI company building the next generation of self-driving technology. • Waabi is growing quickly and looking for diverse, innovative and collaborative candidates who want to impact the world in a positive way. • The Motion Planning team delivers the core module within the autonomy stack that makes decisions and generates trajectories for our self-driving trucks. • As a research engineer for Learnable Planner you will support integration of new AI technologies into our autonomy/planner stack enabling our launch of fully driverless autonomous trucks. • You will contribute towards Waabi's vision of a single AI system that learns end-to-end and in a provably safe manner as well as our revolutionary high-fidelity, closed-loop simulator, Waabi World. • You will integrate cutting-edge ML models in production planning stack from development to validation, deployment, and monitoring. • You will develop necessary interfaces and pipelines in simulation for testing prototype or production planning models. • You will 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. • You will champion engineering excellence, ensuring high-quality, well structured and tested code. • You will stay up-to-date with the latest advancements in AI, machine learning, computer vision, and self-driving technologies.

🎯 Requirements

• 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. • 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. • As we grow, this list continues to evolve!