AI Engineer (Legal AI)

Location: Nashville, Tennessee, United States

Salary: $85,000–$170,000 per year

Company

Law firm operations supported by AI-enabled solutions for attorneys, practice groups, and business departments.

What you will do

• Identify AI opportunities with attorneys, practice groups, and business teams, translating workflows into technical requirements and measurable success criteria.
• Design, develop, test, and deploy AI agents and multi-step workflows using Azure Databricks, Microsoft Azure, n8n, LangGraph, and Databricks Agent Bricks.
• Integrate AI solutions with enterprise applications, APIs, governed data sources, and Microsoft Azure services.
• Develop prototypes and convert successful concepts into reliable, scalable production applications.
• Build evaluation datasets, automated tests, quality measures, and monitoring for agent behavior, retrieval quality, performance, latency, and cost.
• Collaborate with Data Architecture and DevOps teams on deployment, configuration, monitoring, security, documentation, and continuous improvement.

Requirements

• Strong Python programming skills and experience building maintainable production software with APIs, automated testing, version control, and code review.
• Working knowledge of SQL and experience integrating application, data, and AI components.
• Hands-on experience developing AI agents, agent-enabled applications, or multi-step AI workflows involving language models, enterprise information, APIs, or external tools.
• Experience with prompt and context design, structured outputs, tool integration, state management, error handling, and human-in-the-loop workflows.
• Practical experience with Azure Databricks and Microsoft Azure services used for application development, integration, or deployment.
• Knowledge of RAG, embeddings, retrieval techniques, production engineering, enterprise security, access controls, and responsible AI practices.

Nice to have

• Experience with n8n, LangGraph, Databricks Agent Bricks, MLflow, LangSmith, or similar orchestration, tracing, observability, and evaluation tools.
• Advanced Azure Databricks experience with Spark, Delta Lake, Unity Catalog, Lakebase, or Delta Sharing.
• Experience optimizing RAG systems, including vector search, hybrid retrieval, reranking, document processing, and retrieval evaluation.
• Experience with graph databases, knowledge graphs, PostgreSQL, TypeScript, JavaScript, React, MCP, or enterprise authentication models.
• Experience in legal services, professional services, technical consulting, or other environments involving sensitive information and complex workflows.

Culture & Benefits

• Work directly with attorneys, business users, engineers, data architects, and DevOps specialists.
• Participate in code reviews, technical documentation, knowledge sharing, and reusable component development.
• Support AI adoption by gathering user feedback, measuring business impact, and improving deployed solutions.
• Physical requirements include sitting and standing for extended periods and lifting up to 15 pounds.