Tahoe Therapeutics

Tahoe Therapeutics

Senior Machine Learning Engineer

South San Francisco · Senior

Work authorization requiredDetected 232 days ago
Machine LearningDeep LearningTensorFlowPyTorchNLPStatisticsBioinformaticsResearchAnimal Studies

About the role

  • As a Senior Machine Learning Engineer, you will play a leading role in designing the next generation of foundation models of gene regulatory networks powered by Tahoe's large scale single-cell datasets such as Tahoe-100M and beyond.
  • This role is well-suited for someone with a strong background in machine learning and statistics, and an interest in applying cutting-edge breakthroughs in ML to meaningful problems in drug discovery.

Responsibilities

  • We welcome applicants who require visa sponsorship and provide work authorization support for qualified candidates.

Requirements

  • Required
  • Experience in building, testing, training, and deploying modern neural network architectures such as Transformers.
  • Desirable
  • Experience with distributed deep learning using frameworks such as HF Accelerate, Deepspeed, Composer, TorchTitan, Megatron etc.
  • Work together with ML Scientists to develop the next generation of compute optimized models like Tahoe-x1.

Nice to have

  • Solid Engineering and Computer Science fundamentals, ideally with a degree in CS, Math, or equivalent experience.
  • Experience with frameworks like PyTorch, Tensorflow, Keras, JAX and the ability to write hardware optimized code for distributed training
  • Experience with training large neural networks on multiple compute nodes.
  • Stay at the forefront of ML and computational biology research and rapidly adopt state-of-the-art techniques to train AI virtual cell models on our massive single cell datasets.

Skills

  • Exceptional engineering skills to iterate quickly on data processing and training pipelines

Benefits

  • Unlimited Paid Time Off (PTO).
  • US Employees: HMO Kaiser Platinum and PPO Anthem Gold medical as well as vision and dental plans for both the employee and dependents.

Company info

  • About Tahoe Therapeutics Tahoe Therapeutics is a biotechnology company pioneering a fundamentally new approach to drug discovery, one that begins with the biology of real patients.
  • Our Mosaic platform is the first to make in vivo data generation scalable, with single-cell resolution, allowing us to map how drugs affect patient-derived cells in the body across a wide range of biological contexts.
  • We are building the world's largest in vivo single-cell perturbation atlas and using it to train multimodal foundation models that learn the context-dependent nature of gene function, disease progression, and drug response.
  • By combining cutting-edge machine learning with the most biologically relevant datasets ever assembled in drug discovery, our mission is to find better drugs, faster and bring them to more patients who need them.
  • Your role With Tahoe-100M, we solved one of the fundamental bottlenecks in building a virtual model of the cell: generating massive, perturbation-rich, single-cell datasets that capture real biological causality.
  • At Tahoe, we embody a simple philosophy: build in the open, shoot for the moon, and we're looking for people who want to push the frontier of what's possible.

Visa & Work Authorization

  • We welcome applicants who require visa sponsorship and provide work authorization support for qualified candidates.

This listing is sourced directly from Tahoe Therapeutics's careers page and normalized into a canonical job model.