SandboxAQ Inc. logo

AI Sim - Staff ML Research Engineer

Job Overview

Location

Canada

Job Type

Full-time

Category

Machine Learning Engineer

Date Posted

April 1, 2026

Full Job Description

đź“‹ Description

  • • As a Staff ML Research Engineer at SandboxAQ, you will serve as the critical bridge between cutting-edge research and production-grade systems, transforming scientific breakthroughs in Large Quantitative Models (LQMs) into scalable, real-world solutions that accelerate drug and materials discovery—directly impacting global challenges in life sciences and beyond.
  • • You will architect, scale, and optimize the scientific codebases powering LQMs, leading the transition from high-impact research prototypes to robust, production-ready products while ensuring technical excellence across the full software lifecycle.
  • • Day to day, you will: translate scientific papers into scalable ML algorithms and robust code; lead ideation, benchmarking, and execution of complex ML models and datasets; implement advanced software and hardware optimizations for distributed cloud GPU environments; own the full product lifecycle from research to launch and long-term support; collaborate with multidisciplinary scientists and engineers in AI, chemistry, physics, and medicine; drive distributed training pipelines on world-class GPU infrastructure; pioneer hardware-level optimizations to push computational chemistry boundaries; mentor junior engineers and contribute to technical best practices; participate in cross-functional innovation sprints to accelerate model deployment; and ensure seamless integration of ML models into large-scale simulation frameworks.
  • • You will join a global, tech-focused team of experts in AI, chemistry, cybersecurity, physics, mathematics, medicine, and engineering—emerged from Alphabet Inc. in 2022 as an independent, growth capital-backed company—where creativity, collaboration, and impact are deeply cultivated through investment in people and a mission-driven culture.
  • • In this role, you will achieve mastery in scaling research-grade ML systems to production, gain deep expertise in GPU-optimized ML pipelines for scientific computing, and position yourself at the forefront of AI-driven drug discovery—advancing your career while contributing to transformative solutions in life sciences, financial services, and cybersecurity.

🎯 Requirements

  • • MSc (PhD preferred) in Computer Science, Physics, Chemistry, or a related quantitative field with focus on advanced computational methods
  • • 5+ years of industry experience developing productionized software in professional teams
  • • Proven experience training and optimizing large-scale ML pipelines on distributed cloud GPUs (e.g., PyTorch, TensorFlow)

🏖️ Benefits

  • • Competitive base salary, performance-based incentives or bonuses, and equity participation
  • • Comprehensive medical, dental, and vision coverage with generous employer premium contributions, retirement savings with company matching, paid parental leave, and inclusive family-building benefits
  • • Flexible paid time off, company-wide seasonal breaks, and support for flexible work arrangements enabling sustainable performance
  • • Opportunities for continuous learning and growth through on-the-job development, cross-functional collaboration, and access to internal learning and development programs

Skills & Technologies

TensorFlow
PyTorch
Senior
Onsite
Degree Required

Ready to Apply?

You will be redirected to an external site to apply.

SandboxAQ Inc. logo
SandboxAQ Inc.
Visit Website

About SandboxAQ Inc.

SandboxAQ is an enterprise AI company spun out of Alphabet in 2022, focused on applying large quantitative models to solve complex problems in cybersecurity, encryption, sensing, and simulation. Its software and hardware solutions combine AI with quantum-inspired methods to help government and Fortune 500 clients secure data, accelerate materials discovery, and optimize sensing for healthcare and navigation. Based in Palo Alto with global offices, the company partners with large systems integrators and cloud vendors to deploy scalable, physics-aware AI platforms for defense, life sciences, and financial services organizations.

Get more remote jobs like this

Subscribe to the weekly newsletter for similar remote roles and curated hiring updates.

Newsletter

Weekly remote jobs and featured talent.

No spam. Only curated remote roles and product updates. You can unsubscribe anytime.

Similar Opportunities

ARGENTINA
Full-time
Expires Jun 20, 2026
AWS
Terraform
TensorFlow
+4 more

12 hours ago

Apply
Melbourne
Full-time
Expires May 15, 2026
Python
Kubernetes
PyTorch
+4 more

1 month ago

Apply
Heidi Health Pty Ltd logo

Heidi Health Pty Ltd

Melbourne
Full-time
Expires May 15, 2026
Python
Go
TensorFlow
+4 more

1 month ago

Apply
FundraiseUp Inc. logo

FundraiseUp Inc.

Portugal - Remote
Full-time
Expires May 23, 2026
Python
FastAPI
MongoDB
+4 more

29 days ago

Apply