Founding AI/Machine Learning Engineer
San Francisco, CARemoteFULL_TIME28w ago
What You Will Bring Deep Post Training Expertise: Hands on experience with post training models for specific applications (SFT, RLHF, RLAIF, Reward Modeling, Knowledge Distillation, etc) Strong Architectural Foundations: Deep understanding of Transformer architectures (attention mechanisms, positional encodings) and ML
What You Will Bring
- Deep Post-Training Expertise: Hands-on experience with post-training models for specific applications (SFT, RLHF, RLAIF, Reward Modeling, Knowledge Distillation, etc)
- Strong Architectural Foundations: Deep understanding of Transformer architectures (attention mechanisms, positional encodings) and ML systems. And knowing when to use which model (sometimes it's simpler models that win!)
- Large-Scale Training: Experience with distributed training frameworks and optimizing training jobs on GPU clusters.
- High Velocity & Ownership: You thrive in ambiguous environments, learn quickly, and have a bias toward action. You are comfortable shipping in rapid cycles typical of early-stage startups.
- Technical Stack: Proficiency in Python, PyTorch. Familiarity with the modern open-source LLM stacks (HuggingFace, Vertex, Vercel, etc.).
(Healthcare experience is preferred but not strictly required if you have exceptional ML fundamentals.)
What You’ll Work On
Architect the Post-Training Stack: Lead the design and execution of alignment pipelines (SFT, RLHF, RLAIF) that bridge the gap between "exam-passing" models and "clinically useful" systems.
- Leverage Proprietary Data: Utilize our proprietary and open source medical datasets to fine-tune models on edge cases that generic models miss.
- Novel Technique Experimentation: Research and implement cutting-edge post-training methods to optimize model performance, aiming for improvements in calibration and reliability critical for healthcare.
- Safety & Evaluation: Build rigorous evaluation frameworks (LLM-as-a-judge, benchmarks) to detect hallucinations, ensure clinical correctness, and guarantee safety before deployment.
- Strategic Collaboration: Work directly with the co-founders to define the research roadmap and platform strategy.
Bonus Points
- Research Track Record: Published research in high-impact journals or top-tier ML/AI conferences (NeurIPS, ICML, ICLR, CVPR, ACL).
- Top Lab Experience: Background working or interning at top research labs (e.g., FAIR, DeepMind, OpenAI, Google DM, MSR, Stanford/CMU/MIT labs).
- Domain Expertise: Experience dealing with multimodal health data, clinical reasoning, or safety-critical ML systems.
- Entrepreneurial Spirit: You have founded a company, built early-stage products, or enjoy the "zero-to-one" phase of building.
Frequently asked
Is this Founding AI/Machine Learning Engineer role remote?
Yes — this role is remote-friendly (San Francisco, CA).