Prima Mente's goal is to deeply understand the brain, to protect the brain from neurological disease and enhance the brain in health by generating our own data, building brain foundation models, and translating discovery to real clinical and research impact.
Requirements
- Deep understanding of state-of-the-art machine learning methodologies and proven experience in translating them into practical solutions.
- Solid foundation in distributed computing principles, parallel processing, and algorithmic efficiency.
- Experience optimizing ML algorithms for performance, memory efficiency, and compute resource utilization.
- Skilled in designing and implementing scalable data pipelines capable of rapid ingestion, transformation, and processing.
- Deep expertise in modern ML frameworks and tools (e.g., PyTorch, JAX, TensorFlow), and familiarity with state-of-the-art training and inference workflows.
- Demonstrated experience training, optimizing, and deploying large-scale models (7B+ parameters).
- Low level algorithm optimisation
Responsibilities
- Implement high-performance ML algorithms optimised for massive-scale, ensuring reliability, efficiency, and scalability.
- Design, develop, and maintain robust experimentation pipelines enabling rapid iteration, precise evaluations, and reproducible research outcomes
- Refactor and scale prototype research code into clean, maintainable, and performant repositories suitable for production-grade deployments.
- Create high-speed data processing workflows capable of efficiently handling large-scale datasets to accelerate experimentation and model development.
- Experimental design, prioritising high impact experiments with the highest signal:noise ratio.
- running initial experiments with state-of-the-art machine learning models
- optimizing existing code for efficiency and accuracy
Other
- We don’t expect you to check every box. Strong applicants often have depth in some of these and interest in growing into others.
- Skilled in clearly articulating complex ideas, effectively communicating why particular approaches succeed or fail, and providing insightful critical analyses.
- Experience of building highly collaborative research teams.
- Track record of working on hard problems for long periods of time.
- High agency with the ability to jump on any task as needed.
