TriFetchTriFetch

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

RLHFdistributed training

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).

Related