Machine Learning Engineer - Foundation Models for Biology at Prima%20mente

Prima%20mente

San Francisco·4h ago

Mid LevelFull-timeMachine LearningTensorFlowPyTorchAIML

Salary Range

$102,000 - $138,000 /year

Estimated

This salary is estimated based on similar roles. The actual salary may vary.

About the Role

Prima Mente is a frontier biology AI lab seeking a professional to design, implement, and scale foundational AI models and infrastructure for multi-omics at massive scale. This role will drive breakthroughs in scientific understanding and contribute to transformative applications in medicine and biology.

Key Responsibilities

→Implement high-performance ML algorithms optimized for massive-scale
→Design, develop, and maintain robust experimentation pipelines
→Refactor and scale prototype research code into production-grade repositories
→Create high-speed data processing workflows for large-scale datasets
→Design experiments prioritizing high impact with high signal:noise ratio

Requirements

✓Deep understanding of state-of-the-art machine learning methodologies
✓Solid foundation in distributed computing principles and algorithmic efficiency
✓Experience optimizing ML algorithms for performance and resource utilization
✓Expertise in modern ML frameworks (PyTorch, JAX, TensorFlow)
✓Familiarity with large-scale model training and inference workflows
✓Skills in designing scalable data pipelines
✓Ability to articulate complex ideas and provide critical analyses
✓Low level algorithm optimization experience
✓Hardware optimization experience
✓Finetuning optimization experience
✓Experience with large scale data above 2T tokens