Senior AI Engineer

The role

You will design and build agent-driven and RAG-based AI systems to solve legal tasks. You will fine-tune LLMs, develop search pipelines, and implement NLP algorithms for legal documents.

What you will do

• Design and build agent-driven and RAG-based AI systems
• Fine-tune and post-train LLMs using supervised and reinforcement learning
• Develop and optimize large-scale search pipelines
• Implement and refine NLP algorithms to extract and validate information from legal documents
• Run end-to-end evaluations to measure performance
• Work with legal experts to design evaluation protocols and build datasets
• Maintain technical excellence through coding standards and code-review processes

Required qualifications

• Proven expertise in building and managing robust ML pipelines
• Strong Python skills
• Hands-on experience in LLM research and deep learning systems
• Knowledge of search technologies such as full-text search, vector databases, or other large-scale retrieval systems
• Collaborative mindset

Nice to have

• Familiarity with MLOps and integrating ML models into production backends
• Experience in pre-training or deploying large-scale models in real-world scenarios

Benefits

• Hybrid Work
• Health Coverage
• Meal Vouchers
• Performance Bonuses
• Team Retreats & Off-sites

About the company

Lexroom.ai is a platform for legal AI adopted by thousands of legal professionals across Europe. The company aims to make the law faster, clearer, and more reliable.