Machine Learning Researcher
a day ago
Highlight
OnsiteFull-timeEntry
Summary
Prima Mente is seeking a Machine Learning Researcher to join their team in San Francisco, CA. The role focuses on designing, training, and evaluating foundation models that learn from large-scale biological data.
About the Role
As a Machine Learning Researcher, you will work on designing and implementing ML models for biological data, partnering with biologists and clinicians, running end-to-end experiments, and developing evaluation suites for robustness and clinical relevance. You will also perform in-depth model analysis and collaborate on research communication.
About You
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.
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).
Benefits
- Direct patient impact through work on brain disease detection and treatment.
- End-to-end environment from data generation to clinical studies.
- Collaboration with a high-calibre team across ML, biology, and clinical medicine.
- High autonomy and ownership over significant problems.
Prima Mente
The AI Neuroscience company. We're hiring AI Bio, ML Researchers, MLOps, Bioinformatics, SWE & Scientists https://www.primamente.com/join-us/
San Francisco, CA