ML Research Engineer (AI)

TL;DR

ML Research Engineer (AI): Bringing cutting-edge research into production-grade reality for drug discovery with an accent on architecting and scaling scientific codebases for Large Quantitative Models (LQMs). Focus on orchestrating distributed training pipelines on GPU infrastructure and pioneering hardware-level optimizations for computational chemistry.

Location: Remote (United States)

Salary: $134,400 – $252,000 (depending on US tier)

Company

SandboxAQ is a high-growth company delivering AI solutions for life sciences, financial services, navigation, and cybersecurity, emerged as an independent entity from Alphabet Inc.

What you will do

• Translate novel research and scientific papers into high-performing and robust scientific code.
• Lead the ideation, benchmarking, and execution of complex datasets and ML models.
• Integrate ML models into large-scale simulation frameworks to accelerate molecule and medicine creation.
• Drive the full product lifecycle from foundational research and implementation to launch and long-term support.
• Orchestrate distributed training pipelines on world-class GPU infrastructure.

Requirements

• PhD or research-focused MSc in Computer Science, Physics, Chemistry, or a related quantitative field.
• Staff level (5+ years) of industry experience developing productionized software in professional teams.
• Experience or training in data-science tasks related to structural biology.
• Expertise translating research papers into concrete ML software artifacts.
• Experience bridging the gap between research code and product code for external-facing products.
• Must be based in the United States.

Nice to have

• Direct experience in biopharma or training affinity, structure-prediction, or generative chemistry models.
• History of developing and launching successful commercial software products.
• Familiarity with MLOps practices on major cloud platforms.
• Experience in interdisciplinary environments where AI intersects with physical or biological sciences.

Culture & Benefits

• Competitive base salary, performance-based incentives, and equity participation.
• Comprehensive medical, dental, and vision coverage with generous employer contributions.
• Retirement savings with company matching.
• Flexible paid time off, company-wide seasonal breaks, and flexible work arrangements.
• Paid parental leave and inclusive family-building benefits.
• Access to internal learning and development programs and cross-functional collaboration.