Redwood City, California, 94063

Apply now
Full Time
Hybrid
$214000 - $301000/yr

Job Summary

The role involves constructing and sustaining scalable infrastructure essential for the training and deployment of biological foundation models. This position emphasizes optimizing GPU-based data loading, managing extensive distributed systems, and working collaboratively with researchers to identify and resolve training challenges.

Responsibilities

  • Develop and maintain scalable infrastructure for training and deploying biological foundation models.
  • Optimize GPU-native data loading processes.
  • Manage and oversee large-scale distributed systems.
  • Collaborate with research teams to troubleshoot and address training issues.

Qualifications

  • Minimum of 5 years of experience in building machine learning systems or distributed infrastructure at scale.
  • Strong expertise in Python programming.
  • Proficiency in utilizing PyTorch and CUDA.
  • Experience with container orchestration tools such as Kubernetes or Slurm.

Biohub provides equal employment opportunities to all employees and applicants for employment and prohibits discrimination and harassment of any type without regard to race, color, religion, age, sex, national origin, disability status, genetics, protected veteran status, sexual orientation, gender identity or expression, or any other characteristic protected by federal, state or local laws.
To apply for this job, please upload your resume.
supports..pdf, .doc, .docx, or .txt file
X Icon Instagram Icon Facebook Icon LinkedIn Icon
Stay in the Know With STEMHUNTER
Back to Top Machine Learning Engineer, AI
Machine Learning Engineer, AI
close
supports..pdf, .doc, .docx, or .txt file