SandboxAQ

Staff ML Research Engineer

SandboxAQ

full-time

Posted on:

Location Type: Remote

Location: United States

Visit company website

Explore more

AI Apply
Apply

Salary

💰 $177,838 - $306,000 per year

Job Level

About the role

  • Bring content of scientific papers into promising, scalable ML algorithms; and translate these into high-performing and robust scientific code
  • Lead the ideation, benchmarking, and execution of complex datasets and ML models, ensuring seamless integration into our large-scale simulation frameworks.
  • Implement advanced software and hardware optimizations to maximize the efficiency of ML pipelines across distributed cloud GPU environments.
  • Drive software through the entire product lifecycle—from foundational research and implementation to launch and long-term support—ensuring technical excellence at every stage.

Requirements

  • MSc (PhD preferred) in Computer Science, Physics, Chemistry, or a related quantitative field focused on advanced computational methods.
  • Senior (5+ years) 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).
  • Deep familiarity with agentic coding tools (e.g. Claude code, Codex).
  • Experience supporting models in external-facing products, demonstrating the ability to bridge the gap between "research code" and "product code".
  • Direct experience in biopharma or training leading-edge affinity, structure-prediction, or generative chemistry models (highly desired).
  • A history of developing and launching successful commercial software products within a professional engineering team (highly desired).
  • Familiarity with MLOps practices on major cloud platforms to support automated scaling and model monitoring (highly desired).
  • Experience working in interdisciplinary environments where AI intersects with physical or biological sciences (highly desired).
Benefits
  • Comprehensive medical, dental, and vision coverage for employees and dependents with generous employer premium contributions
  • Retirement savings with company matching
  • Paid parental leave
  • Inclusive family-building benefits
  • Flexible paid time off
  • Company-wide seasonal breaks
  • Support for flexible work arrangements that 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
Applicant Tracking System Keywords

Tip: use these terms in your resume and cover letter to boost ATS matches.

Hard Skills & Tools
machine learningML pipelinessoftware optimizationcloud GPU environmentsprogrammingproductionized softwareagentic coding toolsMLOps practicescomputational methodsdata benchmarking
Soft Skills
leadershipcollaborationcommunicationproblem-solvingtechnical excellenceinterdisciplinary teamworkideationexecutionsupportproduct lifecycle management
Certifications
MScPhD