Research Engineer – World Models

Posted 2 days ago

Apply Now

Resume Score

Check how well your resume matches this job before you apply.

Sign in to check score

About the role

  • Research Engineer for Waabi, developing algorithms for world models in autonomous transportation. Collaborating with a team to deliver scalable and efficient AI solutions.

Responsibilities

  • Design, implement, and scale state-of-the-art generative and predictive world-modeling systems:
  • Collaborate closely with Research Scientists to translate cutting-edge model prototypes into robust, large-scale, distributed training and inference pipelines.
  • Optimize model training and inference for efficiency, speed, and reliability on large-scale datasets.
  • Build large scale data pipelines to build high quality datasets for training
  • Ensure the quality, stability, and maintainability of the world model codebase and infrastructure.
  • Stay on top of emerging advances in generative AI, distributed systems, and efficient model deployment in robotics.

Requirements

  • Strong software engineering and implementation: You have very strong Python & PyTorch (or JAX) skills; strong software-engineering fundamentals, and extensive experience with distributed training and large-scale model deployment.
  • Demonstrated technical impact: You have a Master's degree in Computer Vision, Machine Learning, Robotics, or a related field, or equivalent industry experience in model development and scaling.
  • Expert domain knowledge: You have built and deployed generative or predictive models of the physical world, focusing on scale, efficiency, and robustness for real-world applications.
  • Team player: You have worked in a close-knit team of researchers and engineers and have strong communication to deliver successful projects in a fast-paced environment.
  • Bonus: Experience with infrastructure and tooling for large-scale ML training (e.g., cloud platforms, Kubeflow, Ray).
  • Experience with efficient model serving and deployment (e.g., ONNX, TensorRT).
  • Publications or research at top ML/CV/Robotics conference (e.g., CVPR, ECCV, NeurIPS)

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!

Job type

Full Time

Experience level

Mid levelSenior

Salary

$155,000 - $269,000 per year

Degree requirement

Postgraduate Degree

Tech skills

CloudDistributed SystemsPythonPyTorchRay

Location requirements

HybridTorontoCanada

Report this job

Found something wrong with the page? Please let us know by submitting a report below.