CareerPlanSign in

Staff Software Engineer - AI I

India, Bengaluru, Karnataka💼 Full-time🗓 2026-09-17 → 2026-09-25

Core

Lead the development and scaling of enterprise-grade AI systems, including RAG architectures and autonomous agents, for legal, tax, risk, and compliance domains.

Role type

Staff Software Engineer (AI/ML)

Builds

Enterprise AI systems, multi-component pipelines, autonomous AI agents, and scalable model integrations.

Domain

Legal, tax, risk, and compliance technology

Deliverable

production ML models | product features | infrastructure

Required skills

Python, PyTorch, TensorFlow, JAX, MLOps, LLMOps, distributed system design, cloud platforms (AWS/Azure/GCP), Kubernetes, Terraform, vector databases, semantic search

Preferred skills

Agent-based architectures, LLM integration (OpenAI/Anthropic), regulated industry domain expertise, AI governance, publications/patents

Technologies

PyTorch, TensorFlow, JAX, Docker, Kubernetes, Terraform, AWS, Azure, GCP, OpenAI, Anthropic

Responsibilities

Architect scalable AI model integration strategies, lead technical architecture decisions, mentor senior and junior engineers, champion MLOps/LLMOps practices, drive optimization of AI system performance, establish governance frameworks for AI model lifecycle management

Seniority

Staff, hands-on IC with strategic leadership

Sourced via workday · Listed on CareerPlan, which tracks 70,000+ jobs from 20+ sources.