← All roles

AI Engineer

Lexroom· Tech
Apply on company site
Hybrid (Milan)€40k - €100kFull-timePosted 3mo ago

About the role

AI Engineer role at Lexroom.ai focused on designing and building agent-driven and RAG-based AI systems for legal tasks, fine-tuning LLMs, developing large-scale search pipelines, and collaborating with legal experts to create domain-specific datasets.

Responsibilities

  • Design and build agent-driven and RAG-based AI systems to solve complex legal tasks with precision and efficiency.
  • Fine-tune and post-train LLMs using cutting-edge supervised and reinforcement learning techniques to raise response quality.
  • Develop and optimize large-scale search pipelines for LLM-powered agents.
  • Implement and refine NLP algorithms to extract, structure, and validate information from legal documents.
  • Run end-to-end evaluations to pinpoint bottlenecks, measure performance, and guide continuous improvements.
  • Work closely with legal experts to design evaluation protocols and build reliable, domain-specific datasets that strengthen every layer of the stack.
  • Maintain technical excellence by following established best practices, coding standards, and rigorous code-review processes.

Qualifications

  • Proven expertise in building and managing robust ML pipelines.
  • Strong Python skills with 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 and readiness to work with legal experts, researchers, and engineers to keep improving the product.
  • Familiarity with MLOps and integrating ML models into production backends (a strong plus).
  • Up-to-date with LLM innovation, ideally with experience in pre-training or deploying large-scale models in real-world scenarios.

Skills

Required:ML pipelinesPythonLLM researchSearch technologies

Benefits

  • Hybrid Work: Flexibility to balance remote work with in-person collaboration at our Milan HQ.
  • Health Coverage: Comprehensive insurance to keep you secure and focused.
  • Meal Vouchers: Daily support for your lunch breaks.
  • Performance Bonuses: Rewards directly tied to measurable results.
  • Team Retreats & Off-sites: Dedicated time to connect, collaborate, and spark bold ideas.
Last updated Aug 26, 2026. Interested? Apply on the company site.

About Lexroom

Lexroom | AI per Avvocati

Lexroom è il partner AI progettato per amplificare il tuo giudizio professionale: unisce la profondità della ricerca giuridica alla potenza dell'analisi dati, per offrirti precisione, velocità e affidabilità.

AIAnalisi documentaleLexroom per AvvocatiLexroom per AziendeLexroom per CommercialistiLexroom per Consulenti del lavoroLexroom per NotaiLexroom per Pubblica Amministrazione
Company profileWebsite

More open roles at Lexroom