Senior Machine Learning Research Engineer (MLOps)
About The Position
Calico seeks a Senior Machine Learning Research Engineer to join the team building the computational engine that closes the design-test-learn cycle in molecular design and scales the ML infrastructure across the company. Working closely with a biology-fluent group of researchers and engineers, you will build ML platforms to accelerate their research, translate advanced ML methods into scalable code, and iteratively refine these tools as they apply them to real-world drug discovery problems. If you are passionate about building high-impact ML systems, thrive in ambiguity, and are excited to move fluidly between optimizing data pipelines and deploying generative models, this is the role for you. No biology or life sciences background is required for this role.
Requirements
- A strong intellectual curiosity for life sciences
- 5+ years of relevant ML software engineering experience in industry or academia
- Strong software engineering skills, particularly building model training, serving, or evaluation platforms
- Deep expertise in Python and JAX or PyTorch
- Must be willing to work onsite at least four days a week
Nice To Haves
- Large-scale data processing experience (e.g., Ray, Spark, BigQuery) and high-throughput data loading
- Familiarity with common biological datasets and formats (e.g., BAM, TIFF, Zarr, AnnData)
- Hands-on experience with advanced ML architectures (e.g., transformers, diffusion networks), search and optimization methods (e.g., active learning, Bayesian optimization, RL), or large-scale biomolecular models (e.g., AlphaFold)
- Advanced degree in computer science or a relevant field
- Contributions to open-source ML projects or relevant academic publications
Responsibilities
- Build and scale core components of our ML Data, Eval, and Serving platforms, including APIs, libraries, and datastores
- Optimize the training and serving stack to maximize accelerator utilization at scale
- Automate end-to-end ML workflows between computational models and the wet lab, ensuring full reproducibility and lineage tracking
- Optimize and scale our molecular search methods
- Identify opportunities to accelerate model research workflows and ship tools quickly
- Prototype new modeling ideas with researchers and deploy them into the drug discovery pipeline
Benefits
- two annual cash bonuses
Stand Out From the Crowd
Upload your resume and get instant feedback on how well it matches this job.
What This Job Offers
Job Type
Full-time
Career Level
Senior
Education Level
No Education Listed
