Senior Machine Learning Engineer, Diffusion & Reconstruction

Parallel Domain; British Columbia; Hybrid; CAD 150,000 - 180,000; Full time

Job summary

A leading digital twin technology company in Canada is looking for a Senior Machine Learning Engineer to develop innovative machine learning models for 3D and spatiotemporal reconstruction. The role involves optimizing models for performance and productionizing ML components while collaborating with cross-functional teams. Candidates should hold an advanced degree and have deep expertise in ML, especially with frameworks like PyTorch and knowledge of modern 3D representation learning. Competitive compensation of CAD 150,000 - 180,000 per year is offered.

Benefits

Competitive compensation

Collaborative culture

Professional growth opportunities

Qualifications

• Advanced degree in ML, computer vision or related field is required.
• Experience with deep learning frameworks like PyTorch is essential.
• Strong engineering skills are necessary for production code.

Responsibilities

• Design and implement video-to-video diffusion models.
• Optimize models for large-scale environments.
• Develop and ship robust ML components.

Skills

Deep learning frameworks (e.g., PyTorch)

3D vision or reconstruction

Optimization of models and pipelines

Production quality code

Education

MS or PhD in ML, computer vision, robotics

Job description

Parallel Domain is building the world’s most advanced simulation and digital twin platform for autonomy, robotics, and computer vision. Our Replica product creates large-scale, photorealistic digital twins of real-world environments used for testing, validation, and development of autonomous systems.

About the role:

• We are seeking a Senior Machine Learning Engineer to advance the state of learned reconstruction for Replica. In this role, you will drive development of feed-forward and diffusion-based models for 3D and spatiotemporal reconstruction and work closely with cross-functional teams to integrate these models into production systems.

What you’ll do:

• Develop innovative ML models: Design and implement video-to-video diffusion models and efficient 4D feed-forward reconstruction methods.
• Improve scalability and performance: Optimize models and pipelines to support large-scale Replica environments.
• Productionize research: Ship robust, documented ML components integrated with Replica tooling.
• Collaborate cross-functionally: Engage with simulation, rendering, and infrastructure engineers to deliver end-to-end solutions.

What you’ll bring:

• Advanced degree: MS or PhD in ML, computer vision, robotics, or related field.
• Deep ML expertise: Experience with deep learning frameworks (e.g., PyTorch) and large model development.
• Strong engineering skills: Experience taking ML prototypes into production quality code.
• Experience with 3D vision or reconstruction: Demonstrated knowledge of modern 3D representation learning.
• Generative model background: Experience with diffusion models or neural rendering.

What will help you stand out:

• Hands-on experience: Demonstrated experience in developing and deploying production-level machine learning models.
• Research background: Familiarity with academic research in video diffusion models, LoRA fine-tuning, or feed-forward reconstruction.
• Industry knowledge: Understanding of the autonomous systems landscape and the potential applications of machine learning in this domain.
• Publication record: Publications in top-tier conferences or journals related to machine learning.

What we offer:

• Competitive compensation: A base pay range of $150,000 - 180,000/yr, depending on your skills, qualifications, experience, and location.
• Impactful work: The chance to contribute to the advancement of autonomous systems and AI.
• Collaborative culture: A dynamic and supportive work environment where your ideas are valued.
• Professional growth: Opportunities to learn and develop your skills in a cutting-edge field.

If you're passionate about machine learning, 3D reconstruction, generative AI, and the future of autonomous systems, we'd love to hear from you. Apply today and help us revolutionize the world of spatial AI!

This position is available in Vancouver, BC and Karlsruhe, DE.

150000 - 180000 CAD a year