Prima Mente · 3 weeks ago
Machine Learning Researcher
San Francisco
Full-time
Onsite
Mid, Senior LevelPrima Mente is a frontier biology AI lab focused on understanding and enhancing brain health. As a Machine Learning Researcher, you will design and evaluate foundation models that learn from large-scale biological data, collaborating with biologists and clinicians to translate complex questions into machine learning problems.
Artificial Intelligence (AI)BiotechnologyNeuroscience
Responsibilities
Design and implement ML models for large-scale biological data, from pre-training to task-specific fine tuning
Partner with biologists, clinicians, and data scientists to translate biological and clinical questions into tractable ML problems
Run end-to-end experiments: dataset curation, training, evaluation, error analysis, and iteration
Develop and refine evaluation suites for robustness, generalisation, and clinical relevance (e.g. across cohorts, sites, populations)
Explore multi-modal and multi-task training across genomic, epigenomic, transcriptomic, proteomic and clinical signals
Perform in-depth model analysis to extract mechanistic or biomarker-level insights, not just metrics
Collaborate on papers, internal memos, and external communication of key research results
(Optional / plus) Contribute to scaling and optimisation of training and data pipelines, in close collaboration with research engineers
Qualification
Required
Strong background in machine learning or a closely-related field (e.g. deep learning, statistics, optimisation)
Demonstrated experience training and evaluating modern ML models (e.g. transformers, diffusion, graph models, sequence models)
Solid software skills in Python and at least one major ML framework (PyTorch, JAX, or TensorFlow)
Experience designing and running non-trivial experiments: controlling for confounders, building robust baselines, and doing thorough error analysis
Ability to write clearly – whether in code comments, research docs, or papers
Preferred
Experience with large-scale data (e.g. 100B+ tokens or equivalent) or distributed training
Background in computational biology, genomics, epigenomics, neuroscience, or related areas
Work on foundation models (language, vision, or multi-modal) and interest in applying that to biology
Infra/optimisation experience (e.g. FSDP/ZeRO, quantisation, compilation, custom kernels) – especially valuable, but not mandatory
Company
Prima Mente
The AI Biology company.
Founded in 2022
11-50 employees
Funding
Current Stage
Early StageTotal Funding
unknownKey Investors
Innovate UK
2024-03-20Grant
2023-01-01Seed
Company data provided by crunchbase