Senior Machine Learning Engineer jobs in United States
cer-icon
Apply on Employer Site
company-logo

Tahoe Therapeutics · 1 month ago

Senior Machine Learning Engineer

Tahoe Therapeutics is a biotechnology company pioneering a fundamentally new approach to drug discovery. In this role, you will design the next generation of foundation models of gene regulatory networks using large-scale single-cell datasets, contributing to meaningful advancements in drug discovery.

Artificial Intelligence (AI)BiotechnologyTherapeutics
check
H1B Sponsorednote

Responsibilities

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
Work together with ML Scientists to develop the next generation of compute optimized models like Tahoe-x1

Qualification

Machine LearningNeural NetworksData ProcessingPyTorchTensorFlowKerasJAXEngineering FundamentalsStatisticsSoft Skills

Required

Solid Engineering and Computer Science fundamentals, ideally with a degree in CS, Math, or equivalent experience
Exceptional engineering skills to iterate quickly on data processing and training pipelines
Experience in building, testing, training, and deploying modern neural network architectures such as Transformers
Experience with frameworks like PyTorch, Tensorflow, Keras, JAX and the ability to write hardware optimized code for distributed training

Preferred

Experience with distributed deep learning using frameworks such as HF Accelerate, Deepspeed, Composer, TorchTitan, Megatron etc
Experience with training large neural networks on multiple compute nodes

Benefits

Unlimited Paid Time Off (PTO)
Monthly Lunch budget
One-time Office set up budget
US Employees: HMO Kaiser Platinum and PPO Anthem Gold medical as well as vision and dental plans for both the employee and dependents

Company

Tahoe Therapeutics

twittertwittertwitter
company-logo
Tahoe Therapeutics is a biotechnology company that develops AI models of human cells to enhance drug discovery.

Funding

Current Stage
Early Stage
Total Funding
$42M
Key Investors
Amplify Partners
2025-08-11Series A· $30M
2022-12-08Seed· $12M

Leadership Team

leader-logo
Nima Alidoust
Co-Founder and CEO
linkedin
leader-logo
Hani Goodarzi
Co-Founder
linkedin
Company data provided by crunchbase