A Little About Us
UniCourt is a leader in making court data more accessible and useful with our Legal Data as a Service (LDaaS). We provide real-time access to court data through our APIs and online app for business development and intelligence, litigation analytics, litigation tracking, case research, investigations, background checks, due diligence, compliance, underwriting, machine learning models, and process automation.
We provide access to court data from state and federal courts to a diverse list of clients, including Fortune 500 companies and AmLaw firms and industries such as legal, insurance, finance, investigations, government, education, nonprofits, and consumers.
UniCourt is a legal technology company focused on using technology to unlock the potential of legal data. We are based in both California and Mangalore, India and our team includes legal professionals, data scientists, physicists, computer engineers, and sales and marketing, professionals.
About the Job
We are seeking an Associate Technical Lead – AI/ML Engineering with a minimum of 5 years of experience in software development and machine learning. The ideal candidate will have strong hands-on expertise in Large Language Models (LLMs), Retrieval-Augmented Generation (RAG), and MLOps workflows, along with fluency in Python and containerized deployments.
This role involves active contribution to architecture and development, guiding a small group of developers, and working closely with the Technical Lead to deliver scalable AI solutions in production. The Associate Technical Lead is expected to balance individual contributor responsibilities with partial leadership, ensuring high-quality coding, strong collaboration, and technical innovation.
Our company creates some of the world’s most cutting-edge software solutions in the legal industry. We solve difficult problems, work on innovative technology, and build world-class platforms for people and enterprises to interact with court records and other public data sets. With some of the best minds in the industry, we’re one of the most sought-after learning and career destinations in the world of legal tech. If you’re looking to work at a company with opportunities to forge your career path in technology, UniCourt is the right place for you.
Duties & Responsibilities
- LLMs & AI Development
- 1. Develop and deploy applications leveraging Large Language Models (OpenAI, Claude, Llama, DeepSeek, etc.).
- 2. Contribute to fine-tuning workflows, prompt engineering, and inference optimizations.
- 3. Collaborate with the Technical Lead on evaluating models for cost, performance, and scalability.
- Retrieval-Augmented Generation (RAG)
- 1. Implement RAG pipelines by integrating vector databases (FAISS, Pinecone, Weaviate, etc.).
- 2. Optimize embedding generation, retrieval accuracy, and domain adaptation.
- 3. Support experimentation to reduce hallucinations and improve contextual grounding of responses.
- MLOps & Deployment
- 1. Build and maintain CI/CD pipelines for ML workflows using MLflow, Kubeflow, or equivalent.
- 2. Contribute to model monitoring, logging, and retraining strategies.
- 3. Collaborate with DevOps engineers to manage Dockerized deployments in cloud or on-prem environments.
- Software Engineering
- 1. Write clean, well-documented Python code following best practices.
- 2. Build microservices for ML inference and data pipelines.
- 3. Ensure secure, performant, and maintainable system designs.
- Team Contribution & Mentorship
- 1. Mentor junior developers on coding best practices and ML workflows.
- 2. Assist the Technical Lead with code reviews, time estimation, and task assignments.
- 3. Support a team of 3–4 developers/engineers by providing guidance and technical problem-solving.
- Innovation & Research
- 1. Execute Proof-of-Concepts (POCs) with new ML frameworks and open-source tools.
- 2. Stay updated with emerging AI technologies and propose integration opportunities.
Qualifications
- Bachelor’s or Master’s degree in Computer Science, AI/ML, or a related field.
Required Skills
- 5+ years of professional experience in Python software development.
- Hands-on experience with LLMs (fine-tuning, API integration, or open-source model deployment).
- Practical knowledge of RAG pipelines and vector databases.
- Solid experience in MLOps workflows including CI/CD for ML and monitoring.
- Strong understanding of Dockerized deployments and microservices.
- Experience mentoring or guiding junior developers.
- Familiarity with AWS SageMaker or GCP Vertex AI (preferred).
- Strong debugging, problem-solving, and performance optimization skills.
- Good communication and collaboration skills.
Nice to Have Skills & Experience
- Familiarity with distributed training frameworks (PyTorch Lightning, Hugging Face Accelerate).
- Knowledge of data pipeline orchestration (Airflow, Prefect, Dagster).
- Exposure to AI-assisted IDEs (GitHub Copilot, Cursor.io, Windsurf).