AI/ML Research Engineer

Responsibilities

• Implement and optimize machine learning models for protein design
• Build and maintain scalable data processing pipelines for large-scale protein and molecular datasets
• Develop and deploy ML infrastructure for distributed training and inference across GPU clusters
• Collaborate with research scientists to translate experimental ML approaches into production-ready code
• Design and execute ML experiments with clear hypotheses and rigorous analysis
• Optimize model performance and computational efficiency for large-scale protein design tasks
• Build tools and utilities to support rapid prototyping and experimentation by the research team

Required Qualifications

• Bachelor's or Master's degree in Computer Science, Machine Learning, Computational Biology, or related field
• 2+ years of hands-on experience with PyTorch and/or JAX for deep learning applications
• Strong proficiency in Python scientific computing stack (NumPy, Pandas, scikit-learn)
• Experience with distributed computing and GPU optimization techniques
• Familiarity with protein structure analysis, computational biology, or analogous problems in natural sciences
• Understanding of modern deep learning architectures and optimization techniques
• Experience implementing research papers or translating ML approaches to production systems
• Proficiency with version control (Git), testing frameworks, and software engineering best practices
• Strong problem-solving skills and ability to work independently on technical challenges
• Excellent written and verbal communication skills for cross-functional collaboration

Preferred Qualifications

• Experience training LLMs or diffusion generative models
• Knowledge of cloud computing platforms (AWS, GCP) and containerization (Docker, Kubernetes)
• Background in computational biology, bioinformatics, or structural biology
• Experience with large-scale data engineering and ETL pipelines
• Familiarity with MLOps practices and model deployment frameworks

About the role

This is an on-site role and can be based in either Boston, Massachusetts or San Francisco, California. Please only apply if you reside in these cities or are open to relocate.

Pay

Base Salary Range: $140,000-225,000

Schedule

Full-time