AI/ML Engineer (Legal AI)

Location: Hybrid role based in Milan, Italy, with in-person collaboration at the Milan HQ

Annual salary: €45,000–€100,000, plus equity

Company

Legal AI product company building a platform used by legal professionals across Europe to make legal work faster, clearer, and more reliable.

What you will do

• Design and build agent-driven and RAG-based AI systems for complex legal tasks.
• Fine-tune and post-train large language models using supervised and reinforcement learning techniques.
• Develop and optimize large-scale search and retrieval pipelines for LLM-powered agents.
• Implement NLP algorithms to extract, structure, and validate information from legal documents.
• Run end-to-end evaluations, identify bottlenecks, and guide continuous model and product improvements.
• Collaborate with legal experts, researchers, and engineers to create evaluation protocols and domain-specific datasets.

Requirements

• Proven expertise building and managing robust machine learning pipelines.
• Strong Python skills and hands-on experience with LLM research and deep learning systems.
• Knowledge of full-text search, vector databases, or other large-scale retrieval technologies.
• Ability to collaborate with legal experts, researchers, and engineers.
• Strong awareness of current LLM developments, ideally including pre-training or deploying large-scale models in production.
• Commitment to coding standards, best practices, and rigorous code reviews.

Nice to have

• Familiarity with MLOps.
• Experience integrating machine learning models into production backends.

Culture & Benefits

• Autonomous, focused teams working with high ownership and visible impact.
• Hybrid work combining remote flexibility with in-person collaboration in Milan.
• Comprehensive health coverage and meal vouchers.
• Performance bonuses and an equity package.
• Team retreats and off-sites.