Machine Learning Research Engineer

You care deeply about model quality, efficiency, and reliability

3+ years of hands-on experience building and training ML models in PyTorch

Strong Python and software engineering fundamentals, including testing, code quality, and version control

Experience implementing or optimizing transformer-based architectures

Familiarity with cloud infrastructure and containerization (GCP, AWS, Azure, Kubernetes, Docker)

Strong fundamentals in ML, statistics, and/or linear algebra

High-growth opportunity with meaningful impact on the future of protein design

Competitive compensation package with equity participation

401(k) with a strong employer match

Comprehensive benefits including health/dental/vision insurance

Generous PTO policy and commitment to work-life balance

Professional development opportunities in a cutting-edge field at the intersection of AI and biology

Build robust, reproducible and user-friendly pipelines for automated model fine-tuning, alignment and evaluation

Design and implement  modular, easy-to-maintain, multi-model pipelines for protein design.

Develop highly scalable ETL pipelines to process petabyte-scale protein data for model pretraining

Develop software and infrastructure that enable the ML team to work quickly and frictionlessly in distributed and multi-cloud environments

Partner with ML and protein design scientists to prototype research ideas and bring them into production