Staff Research Engineer (ML/Physics Infrastructure)
Мэтч & Сопровод
Для мэтча с этой вакансией нужен Plus
Описание вакансии
TL;DR
Staff Research Engineer (ML/Physics Infrastructure): Building and scaling ML and physics-based infrastructure for drug discovery with an accent on bridging research prototypes and production systems. Focus on optimizing GPU utilization, distributed execution, and creating robust scientific pipelines for AI agents.
Location: Cambridge, MA USA, London, UK, or San Francisco, CA USA
Salary: $224,000 - $294,000 USD
Company
is building Scientific Superintelligence to accelerate discovery in medicine, materials, and energy using AI and proprietary instruments.
What you will do
- Scale and optimize research tools and scientific workflows for production environments.
- Develop ML and physics infrastructure for model training and molecular simulation.
- Improve GPU utilization, distributed execution, throughput, and fault tolerance.
- Architect systems for job orchestration, monitoring, and workflow traceability.
- Package scientific tools into reusable APIs and services for researchers and AI agents.
- Collaborate with computational biophysics and chemistry scientists to productize research.
Requirements
- Strong software engineering skills in Python and experience with ML or scientific computing.
- Experience scaling distributed systems for research, physics, or data-intensive workloads.
- Proficiency with PyTorch, JAX, CUDA, or similar ML/scientific computing frameworks.
- Practical knowledge of Linux, Docker, and reproducible development environments.
- Experience with orchestration tools such as Kubernetes, Slurm, Ray, Flyte, or Argo.
- Ability to translate ambiguous technical needs into robust engineering solutions.
Nice to have
- Familiarity with computational chemistry, biophysics, or drug discovery workflows.
- Experience with cloud GPU infrastructure and multi-cluster execution.
- Experience building tools for LLM agents or automated research workflows.
- Knowledge of workflow observability, checkpointing, and fault-tolerant systems.
- Experience with CI/CD and release practices for research software.
Culture & Benefits
- Competitive base salary with bonus potential and generous early-stage equity.
- Comprehensive medical, dental, and vision coverage (for U.S. employees).
- Flexible time off with generous company-wide holidays.
- Paid parental leave and educational assistance programs.
- Commuter benefits and company-subsidized lunch for office-based employees.
- Dynamic startup environment focused on solving historic scientific challenges.
Будьте осторожны: если работодатель просит войти в их систему, используя iCloud/Google, прислать код/пароль, запустить код/ПО, не делайте этого - это мошенники. Обязательно жмите "Пожаловаться" или пишите в поддержку. Подробнее в гайде →