Member of Technical Staff, Research Engineer

What you'll do

• Build and scale RAG pipelines, agentic workflows, and infrastructure for RL training and reward evaluation.
• Fine-tune or adapt language models for legal tasks, and develop evaluation and monitoring systems to detect regressions and reward hacking.
• Collaborate with product, backend, and legal experts to deliver and improve end-to-end AI features.

What they ask for

• 3+ years of professional software engineering experience building and shipping AI/ML-powered products.
• Deep proficiency in Python and a proven track record of scalable, production-grade systems.
• Familiarity with post-training techniques such as SFT and GRPO, plus hands-on experience with distributed model training, VERL, and vLLM.
• Hands-on experience with LLM APIs, RAG, agentic workflows, and the full ML project lifecycle.

Pay

$200K – $300K USD base salary

Tools

Python, PPO, GRPO, SFT, VERL, vLLM, OpenAI, Anthropic

AI focus

Develop legal AI systems, RAG and agentic workflows, RL training and reward pipelines, model fine-tuning, and evaluation and monitoring for contract review and related tasks.