Founding AI/Machine Learning Engineer
Job description
What You Will Bring
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, Py.
Torch. Familiarity with the modern open-source LLM stacks (Hugging. Face, 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, Deep. Mind, 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.
Description copied from TriFetch's careers page. Read the full posting before you apply.
More jobs at TriFetch
Forward Deployed Engineer
TriFetch· San Francisco, California, USFounding BizOps (Sales & Growth)
TriFetch· Bengaluru, Karnataka, INFounding Engineer (Full-Stack) - India
TriFetch· Bengaluru, Karnataka, IN