Omnilexia is pioneering a new era in legal tech with its innovative AI-powered application, transforming how legal professionals work. Our state-of-the-art system utilizes semantic search to empower and simplify legal research and the drafting of legal documents for law firms and lawyers, providing quick and precise access to crucial information in a user-friendly way.

We’re seeking a highly motivated Senior Machine Learning Engineer to join our team.

You will be at the heart of R&D of our core functionalities, fine-tuning LLMs, creating machine learning models from scratch, delivering RAG and ETL pipelines, with a specific focus on enhancing our legal document search engine.

Highlights:

• Salary B2B: 20-30k net + VAT (but we are flexible if You are an outstanding candidate or a junior)
• Solid employee stock option grant tailored to the stage of our startup
• 100% remote work

In your daily work, you will handle the following:

• Building RAG Pipelines - develop and optimize retrieval-augmented generation pipelines to significantly improve the accuracy and relevance of search results within our legal document retrieval system.
• Developing ETL Pipelines - design and maintain robust and efficient Extract, Transform, Load (ETL) processes for ingesting and processing large-scale textual legal data into various databases, including vector databases.
• Fine-Tuning LLMs - expertly customize and fine-tune large language models to deeply understand and accurately generate legal language, ensuring exceptional performance and reliability in our applications.
• Deploying Models to Cloud Services - lead the implementation and management of machine learning model deployments on cloud platforms, ensuring optimal scalability, robustness, and efficiency.

We are looking for people with:

• 5+ years of proven experience as a Machine Learning Engineer, with a strong and demonstrable focus on natural language processing and information retrieval.
• Exceptional proficiency in programming in Python.
• Extensive hands-on experience with leading machine learning frameworks such as TensorFlow or PyTorch.
• In-depth understanding of Retrieval-Augmented Generation (RAG) techniques and a strong track record of building and optimizing related pipelines.
• Proven expertise in developing and optimizing ETL pipelines for handling large volumes of data.
• Significant hands-on experience in fine-tuning and successfully deploying Large Language Models (LLMs).
• Strong familiarity with major cloud platforms (AWS, Azure, or Google Cloud) and substantial experience in deploying machine learning models within these environments.
• Comprehensive knowledge of vector databases and practical experience in their integration into machine learning pipelines.
• English language skills enabling fluent work and communication with the team (the communication in Engineering team is in English)
• “Get things done” attitude

Preferred qualifications: 

• Familiarity with the Legal Tech industry
• MLOps experience
• Open source contributions
• Academic publications (can be arXiv)

Why You'll Love Working With Us:

• A fast paced environment where You can thrive and see customers using solutions You deliver from day one.
• Employment based on a B2B contract.
• Flexible working hours, focused on deliverables - seriously, we don’t care if You have a hairdresser’s appointment or go to the gym at 1PM as long as You are reachable on Slack and ship Your work on time and are willing to catch up in the evening
• 100% remote work.
• Salary B2B: 20-30k net + VAT (but we are flexible if You are an outstanding candidate or a junior)
• Solid employee stock option grant tailored to the stage of our startup
• Flexible paid time off, you get 20 days of annual leave + bank holidays.
• Benefits: Luxmed, we will pay for Your online courses plus we can give You gym allowance or whatever You need to make You happy and determined to work (as long as it’s not a brand new S1000RR)
• Soon: possibility to work in Warsaw office (only if You want to)

Do not apply if:

• You are not eager to grow, develop new skills and learn new techniques
• You are not curious
• You are not relentless in solving problems
• You want to repetitively do the same things
• You are not laser focused on customer value delivery
• You do not want to be a part of a team’s hustle and just want to do your work and go home

Definitely apply if:

• You are slightly underqualified for the position but are drastically determined to quickly catch up
• You lack a degree but can demonstrate expertise in the ML field (we buy skills, not diplomas)
• You lack substantial commercial experience but can demonstrate knowledge (actual completed projects, not copy-paste Udemy ones) and passion for AI (we may offer You a junior position if we see a potential for You to quickly start delivering like a senior under our mentorship)
• You love to be on the forefront of AI research and want to integrate newest advancements in the field into actual product
• You want to be a part of a supportive environment that cherishes every success together
• You want to be recognized for outstanding performance
• You find a purpose in Your work
• You want to be part of company-organized “workations”  

Hiring process:

• Online coding task
• General technical interview
• Live coding interview
• As part of recruitment process You can talk to any of our employees to get references on how it is to work at Omnilexia

About us 

We’re a small but exceptionally ambitious team of engineers, designers, and product thinkers with a vision to transform legal research space through AI. Most of us combine multiple areas of expertise — from design & product, to ML & backend, to full stack & DevOps — or are actively stepping outside their comfort zones to grow into new roles. We believe in a shared sense of purpose, and approach problems with creativity, curiosity, and a healthy disregard for how things are “usually done.”

We value smart work over hard work — we optimize for clarity, impact, and long-term maintainability. Of course, we’re not afraid of rolling up our sleeves when things get tough, but we’d rather solve problems cleverly than brute-force them. We promote ownership and despise micromanagement to allow everyone to grow as fast as they possibly can. Furthermore, we value transparency, so you can count on honest feedback from us, and we welcome the same from you.

Even though we’re a fully remote team with a low-meeting culture, collaboration is at the heart of how we work. We support each other in real time, hop on quick calls when it makes sense, and frequently practice pair programming to tackle tricky challenges together. Our team genuinely enjoys working here — and it’s reflected in our collaboration, mutual respect, and the passion we bring to everything we build.