O

Onit

AI-driven enterprise legal management platform

AI Engineer

AI Application EngineeringConfirmed live in the last 24 hours
Full-timePune, Maharashtra, IndiaNo salary listedIn-person

About the Job

Requirements

  • Strong software engineering background (Python preferred) and experience shipping backend services
  • Deep hands-on experience building agentic LLM systems from first principles: agent loops, tool interfaces, planning/replanning, memory/state, and failure handling
  • Strong context engineering ability: retrieval strategies, routing, grounding, context budgeting, and long-context tradeoffs
  • Strong evaluation discipline: golden datasets, regression gating, automated eval pipelines, and online monitoring
  • Practical experience with LLM APIs (OpenAI/Azure OpenAI/Anthropic/Gemini) and AI orchestration frameworks
  • Excellent debugging, systems thinking, and problem decomposition skills
  • Comfortable operating in fast-paced, ambiguous environments with high ownership
  • You’ve shipped an LLM/agent system in production and can clearly explain the failure modes you discovered, the evals you built to catch regressions, how you improved cost/latency while increasing quality, how you monitored and iterated safely over time
  • You keep up with industry developments (model releases, frameworks, best practices) and can translate them into pragmatic improvement

Responsibilities

  • Build and integrate AI-driven features using LLM APIs (OpenAI / Azure OpenAI, Anthropic, Gemini on Vertex AI)
  • Design and implement tool-using agents (structured function calling, schema validation, retries, fallbacks)
  • Build multi-agent workflows when appropriate (e.g., planner/worker, reviewer/critic, specialist routing) and know when a simpler architecture is better
  • Create agentic workflows such as document understanding, extraction, reasoning over evidence, task automation, and multi-step decision support
  • Own context engineering end-to-end: dynamic context assembly (retrieval + state + tool outputs); context budgeting and compression/summarization; grounding strategies to reduce hallucinations and improve consistency
  • Implement retrieval-augmented generation (RAG) and search workflows using off-the-shelf vector stores and embedding services
  • Establish evaluation frameworks for accuracy, reliability, and output quality
  • Build task-specific eval suites: golden datasets, adversarial cases, regression tests, and rubric-based scoring
  • Set up automated evaluation pipelines and release gates (CI/CD-friendly) tied to prompt/model/version changes
  • Define and monitor online metrics (e.g., task success rate, human override rate, safety flags, latency, cost) and run experiments/A-B tests where appropriate
  • Use LLM-as-judge responsibly: calibrate, validate, and pair with human labels when needed
  • Develop scalable backend services and APIs that incorporate AI functionality
  • Integrate AI pipelines into existing cloud, microservices, and event-driven architectures
  • Implement observability and analytics for all AI features (tracing, evaluations, prompt versioning, cost tracking) Example tooling: Langfuse (and/or OpenTelemetry-compatible stacks)
  • Ensure reliability, uptime, performance, and security of AI services
  • Build internal tooling for evaluation, testing, prompt/version management, and safe deployment
  • Partner with product managers, designers, the Chief Architect, and domain SMEs to shape AI-first solutions
  • Rapidly prototype concepts and iterate based on user feedback and measurable eval results
  • Translate business problems into well-structured AI workflows without requiring ML model training
  • Document system behavior, known failure modes, and operational playbooks
  • Implement guardrails, checks, and fallback logic for safe and predictable AI behavior
  • Help define and follow compliance, privacy, and responsible AI guidelines
  • Design for safe tool execution (bounded actions, permissions, escalation paths, human-in the-loop review)

Desired Qualifications

  • Experience with cloud platforms (AWS and/or GCP), microservices, and event-driven systems
  • Experience with observability stacks (OpenTelemetry, Datadog, Honeycomb) and AI-specific tooling (e.g., Langfuse, Braintrust, HumanLoop, W&B Weave)
  • Experience with workflow orchestration for long-running jobs (Temporal, Celery, Airflow)
  • Experience building enterprise AI features (permissions, auditability, compliance constraints)
  • Experience with safety/policy layers (PII handling, prompt injection defenses, sandboxed tool execution)

About the Company

Onit

AI-driven enterprise legal management platform

Onit provides AI-native enterprise legal management and legal operations software for corporate legal departments, legal teams, and law firms. Its platform unites core legal ops capabilities—legal spend management, matter management, e-billing, contract lifecycle management, intake and workflow automation—and adds AI-driven review of invoices, contracts, and matters into a single connected system. The product is delivered as a SaaS subscription and now includes AI agents and generative AI across spend, contract, and workflow tools. What sets Onit apart is a platform built specifically for legal needs, designed to scale for regulated industries, and integrated with AI across multiple modules to streamline operations in one environment. The goal is to help enterprise legal teams run their operations more efficiently, control costs, shorten contract cycles, and improve compliance and governance.

Company size
501-1,000
Company stage
Growth Equity (Venture Capital)
Total funding
$226.4M
Headquarters
Atlanta, GA
Founded
2009

Recently Posted Jobs

Explore more opportunities at Onit

See more jobs