Senior AI Engineer, Legal LLMs - Remote, Flexible Hours

AllyNd Partners; Chicago (IL); Remote; USD 120,000 - 160,000; Full time

Benefits offered by this job

Annual learning stipend

Flexible hours

Comprehensive health benefits

GPU compute access

Job summary

An innovative SaaS company in Chicago is looking for an experienced AI Engineer specializing in LLMs to develop next-generation legal intelligence tools. The candidate will fine-tune models, design RAG pipelines, and evaluate AI features in a fast-paced startup environment. Strong experience in machine learning, NLP, Python, and tools like PyTorch is essential. The position offers a fully remote setup with flexible hours and comprehensive health benefits.

Qualifications

• 7+ years in ML/AI engineering roles with a focus on NLP or LLMs.
• Strong experience with model fine-tuning and RAG architectures.
• Proficiency in Python and deep learning frameworks.

Responsibilities

• Fine-tune and deploy LLMs for legal document analysis.
• Design and implement Retrieval-Augmented Generation (RAG) pipelines.
• Optimize inference workflows for latency and accuracy.

Skills

Model fine-tuning

NLP experience

Python proficiency

Deep learning frameworks

Deployment experience

Tools

PyTorch

JAX

Hugging Face

Pinecone

Weaviate

FAISS

Job description

An innovative SaaS company in Chicago is looking for an experienced AI Engineer specializing in LLMs to develop next-generation legal intelligence tools. The candidate will fine-tune models, design RAG pipelines, and evaluate AI features in a fast-paced startup environment. Strong experience in machine learning, NLP, Python, and tools like PyTorch is essential. The position offers a fully remote setup with flexible hours and comprehensive health benefits.