Software Engineer, ML Infrastructure
Core
Own the backend of the Model Performance Dashboard and design evaluation tools for virtual cell model performance.
Role type
Senior IC ML Infrastructure Engineer
Builds
Production data and ML platform, single-cell data pipelines, lab-in-the-loop infrastructure, train-and-evaluate CI system
Domain
Biotech / Single-cell genomics / ML Infrastructure
Deliverable
production ML models
Required skills
End-to-end ML platform ownership, AI agent output verification, Python, PyTorch, distributed training diagnostics, GPU utilization optimization, stakeholder collaboration, code review, system design
Preferred skills
Single-cell or perturbation data (AnnData, h5ad), model evaluation harnesses, multi-node distributed training (DDP, FSDP, NCCL, SLURM), engineering leadership
Technologies
Python, PyTorch, AnnData, h5ad, DDP, FSDP, NCCL, SLURM
Responsibilities
Design and build evaluation tools for model comparison; Redesign and operate the single-cell data pipeline; Own lab-in-the-loop infrastructure; Build validation frameworks for agent-written code; Own the train-and-evaluate CI system; Work directly with ML researchers and scientists to define requirements
Seniority
Senior, hands-on IC
