ML Engineer
Apply NowCompany: Topos Bio
Location: San Francisco, CA 94112
Description:
We're seeking an ML Engineer to build generative models using large-scale biological simulation data.
Topos Bio
The cell is the fundamental unit of life. Conversely, every disease is the consequence of the breakdown of some cellular machinery.
At Topos Bio, we're building tools to simulate cellular building blocks and then using these simulations to design therapeutics for a variety of diseases.
We're first setting our sites on neurodegenerative diseases. These diseases affect 10s of millions of people and currently have no cure. The machinery implicated in these diseases has long been considered "undruggable". We believe we can change this.
Challenges you'll tackle
Topos Bio
The cell is the fundamental unit of life. Conversely, every disease is the consequence of the breakdown of some cellular machinery.
At Topos Bio, we're building tools to simulate cellular building blocks and then using these simulations to design therapeutics for a variety of diseases.
We're first setting our sites on neurodegenerative diseases. These diseases affect 10s of millions of people and currently have no cure. The machinery implicated in these diseases has long been considered "undruggable". We believe we can change this.
Challenges you'll tackle
- Develop and optimize generative models trained on large-scale simulation datasets
- Implement RL and fine-tuning techniques to improve model performance
- Architect and deploy scalable training pipelines for high-throughput ML experimentation
- Engineer efficient data preprocessing and augmentation strategies
- Implement and adapt state-of-the-art ML methods from research papers into production-ready systems
- Collaborate with AI researchers, computational scientists, and engineers to integrate ML models with simulation workflows
- Balance rapid prototyping with scalable, production-grade ML solutions
- Communicate technical trade-offs and novel findings to interdisciplinary teams
- Strong background in generative modeling, reinforcement learning (RL), and fine-tuning modern ML models
- Experience implementing and optimizing ML methods from research papers using frameworks like PyTorch or TensorFlow
- Hands-on expertise with large-scale data pipelines, distributed training, and cloud-based ML infrastructure
- Experience with protein modeling, molecular dynamics, or bioinformatics
- Familiarity with HPC and cloud-based ML training
- Pharma/biotech experience, ideally with a focus on drug discovery
- Competitive salary and equity package
- Comprehensive health insurance coverage for you and your family