Staff ML Engineer, Applied AI

TIFIN

138d•$190,000 - $225,000

About The Position

As a Staff ML Engineer on the Applied AI team, you will own and lead projects across our various generative AI product lines. We are looking for individuals who have worked end-to-end and have experience building generative AI products where LLMs are a core component. Expertise with fine-tuning and reinforcement learning is critical. This is an individual contributor and technical leadership role reporting into the Head of AI.

Requirements

5+ years of experience (post-BS/MS) in applied AI/ML engineering
Experience shipping generative AI products powered by LLMs: designing and implementing agentic workflows that enable autonomous AI behaviors, leveraging the latest state-of-the-art research
Hands-on experience in various LLM fine-tuning techniques (e.g. LORA), LLM inference frameworks (e.g. vLLM), advanced RAG pipelines
Solid expertise in reinforcement learning fine tuning methodologies and frameworks to optimize AI models
Resilience and adaptability - experience working at early-stage startups is a plus

Responsibilities

Design and fine-tune open source and proprietary LLMs for various tasks such as answering questions, summarization, reasoning and planning, etc.
Build advanced Retrieval Augmented Generation (RAG) pipeline including rewriting, embedding fine-tuning, hybrid search, reranking, knowledge graphs, etc.
Experience shipping generative AI products powered by LLMs: designing and implementing agentic workflows and multi-agent systems that enable autonomous AI behaviors, leveraging the latest state-of-the-art research
Apply reinforcement learning techniques—including reinforcement learning fine tuning (e.g., PPO, DPO, GRPO)—to continuously optimize model performance.
Implement a comprehensive evaluation framework and metrics for model performance
Deploy models into production environments and ensure low latency, reliability, and scalability.
Collaborate with product team and software engineering team to build end-to-end product systems.