SandboxAQ Inc. logo

AI Sim - Staff ML Research Engineer

Job Overview

Location

United States

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, architecting and scaling the scientific codebases that power Large Quantitative Models (LQMs) to revolutionize drug and materials discovery. Your work will directly enable in silico design to become the dominant paradigm in pharmaceutical innovation, accelerating the development of life-saving medicines through advanced ML and physics-based modeling.
  • • You will lead the translation of scientific research into robust, scalable ML algorithms by architecting and implementing high-performance code that transforms academic prototypes into production-ready systems, ensuring technical excellence across the entire software lifecycle from foundational research to long-term support and maintenance.
  • • Your day-to-day responsibilities will include driving the ideation, benchmarking, and execution of complex datasets and ML models, implementing advanced software and hardware optimizations for distributed cloud GPU environments, and owning the full lifecycle of LQM-powered simulation frameworks to maximize efficiency and scientific impact.
  • • You will collaborate within a multidisciplinary team of experts in AI, chemistry, physics, mathematics, and engineering, working in a culture that values creativity, ownership, and transformative impact, where your contributions will directly advance SandboxAQ’s mission to solve global challenges through AI-driven scientific innovation.
  • • In this role, you will deepen your expertise in distributed ML systems, GPU optimization, and agentic coding practices while gaining hands-on experience in bringing research breakthroughs to commercial products, positioning you at the forefront of AI-accelerated scientific discovery and enabling significant career growth in a high-impact, fast-growing technology company.

🎯 Requirements

  • • MSc (PhD preferred) in Computer Science, Physics, Chemistry, or a related quantitative field with a focus on advanced computational methods
  • • 5+ years of industry experience developing productionized software in professional engineering teams
  • • Proven experience training and optimizing large-scale ML pipelines on distributed cloud GPUs using frameworks such as PyTorch or 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 to enable 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