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 You will...
Build standardized distributed training frameworks for research and production, drive our training towards new levels of stability and efficiency.
Comprehensively profile model runtime and memory to pinpoint performance bottlenecks.
Identify and evaluate emerging technologies that can be adopted into Waabi’s training and inference frameworks. Examples include designing new CUDA kernels, quantization-aware training and inference, and compilation/deployment techniques.
Work with researchers and ML engineers on best-practices for optimal resource usage.
Create and improve tooling and dashboards to ensure broad adoption of your work.
Qualifications:
MS/PhD or Bachelors degree with a minimum of 4 years of industry experience in Computer Science, Robotics and/or similar technical field(s) of study.
Solid coding proficiency in a variety of coding languages including Python, C++ or Rust.
Experience in deep learning frameworks such as PyTorch or Jax.
Skilled in profiling CPU and GPU code using tools such as PyTorch Profiler and NVIDIA Nsight.
Open-minded and collaborative team player with willingness to help others.
Passionate about self-driving technologies, solving hard problems, and creating innovative solutions.
Bonus/nice to have:
Experience in identifying when custom CUDA kernels are needed, and implementing them.
Experience in Bazel in a monorepo environment, and integrating third party packages into dev environments.
Experience with Kubernetes-based training platforms.
The US yearly salary range for this role is: $184,000 - $272,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:
Waabi provides a competitive benefits package that includes:
Competitive compensation and equity awards.
Health and Wellness benefits encompassing Medical, Dental and Vision coverage.
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.
World-class facility that includes a gym, games room (ping pong table, video game consoles, board games, etc), multiple collaborative working spaces and a gorgeous patio!(when in office)
As we grow, this list continues to evolve!
Waabi is an equal opportunity employer that celebrates diversity and is committed to creating a supportive, inclusive, and accessible environment for all employees. If reasonable accommodation is needed to participate in the job application or interview process please let our recruiting team know.
You apply on the employer's page; AtiveSite does not receive or process applications and is not affiliated with the employer. The description above is shown as the source published it.
We could not match this job title to one occupation in the official wage statistics, so no pay estimate is shown. Pay is not in the source listing; ask the employer for the range and compare in the 240 job niches.
How to prepare and behave
This is editorial guidance, not information from the employer.
Read the whole description and match your examples to its requirements; use the same words the posting uses where they are true for you.
Research the employer before the interview: what it does, who its customers are and recent news; prepare two or three questions about the work.
Arrive early (or log in early for a video call), dress one step above the daily dress code, and test your camera and audio beforehand.
Answer with short stories: the situation, what you did and the result. Give numbers when you have them and never invent them.
Be honest about gaps; say what you did to learn and how fast you can close them.
Ask about the team, the first 90 days, how success is measured and the schedule; do not open with salary and time off.
When pay comes up, know the market range for the occupation and your lowest acceptable offer before you answer; ask for the full package in writing.
Send a short thank-you message within a day, mentioning one thing you discussed.
The listing mentions a remote or flexible location. Confirm with the employer whether remote work applies to your state: taxes, licensing and benefits can depend on where you live.
Have code, data work or projects you can show, and be ready to think aloud: many technical interviews test how you reason, not only the answer.
The listing is remote or has no US state we could identify. Read the country guide for pay, taxes, law and cost of living by state.
Where to live
The job city has no coordinates we can place on the map, so no housing comparison is shown. Use the 250 largest cities table to compare rent and income.
Career paths
We could not match this title to one occupation, so here are the other sub-niches in Technology, where your skills may carry over:
This job has no single city we can place on the map (remote, flexible or unmatched location), so no distance table is shown. Browse all Technology jobs with distances from any of the 20 largest cities.
We do not hold headcount or revenue for this employer. Check its careers site and, if it is listed on a stock exchange, its public filings (SEC EDGAR search).
What it is like to work at a company (culture, schedules, benefits) is not in the source data, so we do not describe it. Read the description and ask the questions under "How to prepare".
What is the Senior / Staff ML Training Optimization Engineer job at Waabi?
A private-sector role in technology (data, analytics and ai) listed 186 days ago. The full description is on this page and on the employer's page.
Where is this job?
Remote / flexible (United States). A remote or flexible location is listed; confirm with the employer which states are accepted.
How do I apply?
Use the "View this job and apply on the employer's page" button. AtiveSite does not receive applications.
How much does it pay?
The listing does not publish pay, and we could not match the title to one official occupation.
What is the cost of living there?
See the state or country guide for rent, income and taxes.
How old is this listing?
It was posted on 2026-05-08 (186 days ago). Jobs older than 90 days are not shown.
Are there similar jobs?
Yes: browse data, analytics and ai jobs in the United States.
Sources and terms
Job data: Waabi careers page public feed (listing dated 2026-05-08). The job disappears from this site when it disappears from the source.
Wages and projections: BLS OEWS (May 2025) and Employment Projections 2025–35.
City and county figures: Census Bureau ACS 2020–2024, BEA, BLS QCEW and LAUS, NASA POWER climatology; distances from Census Gazetteer coordinates.
AtiveSite is not affiliated with or endorsed by the employer or any government agency, and does not process applications. Verify every detail with the employer before acting.
