AI Engineer - Trellis

About Trellis Law:

Trellis is the leading provider of state trial court data in the U.S. We leverage AI and Machine Learning to analyze hundreds of millions of state trial court documents, transforming complex data into actionable insights. Our platform empowers attorneys, law firms, and corporate legal departments with data-backed intelligence for faster, higher-quality decisions on judicial rulings, litigation strategy, and competitive analysis. Founded in 2018, Trellis has experienced rapid growth and is now trusted by many of the nation’s largest law firms and corporate legal teams. Come join Trellis 's mission to make state trial court data searchable, accessible, and impactful through a single, powerful interface.

About The Role:

We are now looking for a seasoned, impact-driven AI/ML Engineer that wants to be a meaningful part of our Engineering team. You will be building clean, scalable architecture to keep up with our rapidly growing user base and data. As an AI/ML Engineer at Trellis Law, you will work on backend services, data flows and analytics. You will organize, transform and serve large data sets in a high speed and high volume environment.

Key Responsibilities:

• Use your experience to design features connecting natural language queries with a large corpus of legal knowledge.
• Build a data architecture you are proud to highlight.
• Use unstructured data to build large scale data sets.
• Work on a team dedicated to ML and Data Science with ownership of multiple projects and products.
• Collaborate with the Product team to understand new features.

About You:

• At least 7 years of backend engineering experience, at least 2-3 of which were spent building products using LLMs.
• At least 1-2 years of experience fine-tuning LLMs for domain-specific and otherwise custom use-cases.
• At least 1-2 years of experience building or using tooling to evaluate LLMs and LLM-based products, e.g. experience evaluating extractive and abstractive summaries.
• At least 4-5 years of experience deploying custom machine learning models and working on end-to-end ML pipelines, including with unstructured data.
• Experience building semantic search is a plus.
• Python proficiency and working knowledge of SageMaker and Bedrock.
• Strong understanding of Machine Learning, specifically Large Language Models.
• Have mentored other Engineers in multiple projects.
• Startup experience is highly valued.
• BS/MS in Computer Science or a related technical discipline, or equivalent technical work experience.

Benefits:

• Flexible remote-first work culture (with office space in Los Angeles).
• We cover 100% of health, dental, and vision insurance for you and for spouses and dependents.
• We have a 401(k) retirement sa