AI Tech Lead - Staff Machine Learning Engineer

Responsibilities

• Lead technical evaluation and adoption of agentic AI platforms including Anthropic, LangChain, and AWS Bedrock
• Architect and productionize multi-agent AI systems for detection, triage, and investigation workflows
• Own core agent architecture including planning, tool orchestration, memory, and context engineering
• Lead AI agent evaluation systems using offline/online pipelines and synthetic data generation
• Drive LLM fine-tuning and alignment for security and observability use cases
• Design scalable LLMOps infrastructure including inference routing and latency optimization
• Partner with product and security leadership to deliver end-to-end AI agent capabilities
• Provide technical direction and mentorship for AI engineers

Requirements

• B.Tech, M.Tech, or Ph.D. in Computer Science, Machine Learning, or Data Science
• 5+ years of industry experience building and leading production ML/AI systems
• Strong foundation in machine learning, distributed systems, and large-scale system design
• Deep understanding of LLMs, prompt engineering, and agentic AI design patterns
• Strong proficiency in Python and modern ML/AI ecosystems
• Experience designing and operating evaluation frameworks for ML/LLM systems
• Proven ability to lead complex technical initiatives across teams

Preferred Qualifications

• Hands-on experience scaling agentic AI or multi-agent architectures in production
• Experience with LangGraph, LangChain, CrewAI, or similar frameworks
• Experience with Anthropic, OpenAI, AWS Bedrock, or Vertex AI
• Experience with LLM fine-tuning pipelines (SFT, RLHF/RLAIF)
• Strong background in LLMOps and inference optimization
• Experience with PyTorch, MLflow, Airflow, Docker, and Kubernetes
• Experience applying AI/ML to security or observability data

About the Company

Sumo Logic enables customers to deliver reliable and secure cloud-native applications through its SaaS Analytics Log Platform, helping practitioners ensure application reliability and protect against modern security threats.