Multiple Roles at Prior Labs
Core
Building foundation models for structured data (tables) to advance clinical trials, financial models, and scientific experiments.
Role type
Senior ML Infrastructure Engineer / Research Scientist, Foundation Model / Research Engineer, Foundation Model / ML Engineer, Cloud Platform / Full Stack Engineer, ML Platform
Builds
Pre-trained transformer models (TabPFN) and the multi-cluster GPU infrastructure supporting them.
Domain
Deep learning, foundation models, causal structure, scientific computing
Deliverable
production ML models | infrastructure
Required skills
GPU cluster management, Slurm, multi-cloud infrastructure, training performance optimization, novel model architecture design, scaling laws, experiment design, training and evaluation infrastructure, Python, FastAPI, Terraform, Kubernetes, TypeScript, React, Postgres
Preferred skills
Experience with large-scale distributed training, causal inference, multimodal learning
Technologies
GCP, Slurm, Python, FastAPI, Terraform, Kubernetes, TypeScript, React, Postgres
Responsibilities
Own multi-cluster GPU infrastructure and tooling layer; drive model agenda with novel architectures and scaling; design experiments and co-author papers; design and scale backend for model serving and finetuning; build product end-to-end
Seniority
Senior, hands-on IC / Principal, strategy & mentorship