[Remote] AI Tech Lead - Staff Machine Learning Engineer

Note: The job is a remote job and is open to candidates in USA. Sumo Logic, Inc. empowers modern digital businesses by offering a cloud-native applications analytics platform. They are seeking an AI Tech Lead to design and deliver advanced AI systems for their Security Operations Center, focusing on the development of scalable multi-agent architectures for real-time data processing and security operations.ResponsibilitiesLead and partner with fellow leadership members and teams on technical evaluation and adoption of cutting-edge agentic AI platforms, including Anthropic (Claude), LangChain/LangGraph, AWS Bedrock, and other emerging agent frameworksArchitect, prototype, and productionize multi-agent AI systems for Agentic SOC use cases, including detection, triage, investigation, and response workflowsOwn the design of core agent architecture components, including planning, execution, tool orchestration, memory, context engineering, and long-running agent workflowsLead AI agent evaluation systems, including offline and online evaluation pipelines, golden datasets, synthetic data generation, human- and LLM-based judging, and continuous quality monitoringDrive LLM fine-tuning and alignment efforts to improve domain-specific reasoning, accuracy, and reliability for security and observability use casesDesign scalable LLMOps and AI agent infrastructure, including inference routing, latency optimization, cost control, and production observability for agent systemsPartner with product, security, and data platform leadership and teams to deliver end-to-end AI agent capabilities from prototype to customer-facing production systemsLead and partner on technical direction and mentorship for AI engineers working on agentic AI and LLM systemsDefine and implement best practices for AI safety, reliability, evaluation, and monitoring in production agentic systemsOperate as a senior technical owner in ambiguous problem spaces—setting technical direction, breaking down complex problems, and driving delivery across teamsSkillsB.Tech, M.Tech, or Ph.D. in Computer Science, Machine Learning, Data Science, or a related technical field5+ years of hands-on industry experience building, operating, and leading production ML/AI systems, with demonstrated technical leadership and ownershipStrong foundation in machine learning, distributed systems, data pipelines, and large-scale system designDeep industry understanding of LLMs, prompt engineering, context engineering, agentic AI design patterns, and reasoning workflowsStrong proficiency in Python and modern ML/AI ecosystemsExperience designing and operating evaluation frameworks for ML/LLM systems (offline + online)Proven ability to lead complex technical initiatives across teams and influence architecture decisionsExcellent communication skills and ability to translate complex AI systems into business impactHands-on experience building and scaling agentic AI systems or multi-agent architectures in productionExperience with modern agent frameworks such as LangGraph, LangChain, CrewAI, or similarExperience with major foundation model platforms such as Anthropic, OpenAI, AWS Bedrock, or Vertex AIExperience with LLM fine-tuning pipelines (SFT, RLHF/RLAIF, preference learning, domain adaptation)Strong background in LLMOps, including inference optimization, latency/cost management, observability, and production monitoringExperience with ML infrastructure and tooling such as PyTorch, MLflow, Airflow, Docker, Kubernetes, and cloud platforms (AWS/GCP/Azure)Experience applying AI/ML to security, observability, or large-scale log/telemetry data is a strong plusBenefitsIn addition to base pay, certain roles are eligible to participate in our bonus or commission plans, as well as our benefits offerings and equity awards.Company OverviewSumo Logic is a provider of cloud-based machine data analytics that enables reliable and secure cloud-native applications. It was founded in 2010, and is headquartered in Redwood City, California, USA, with a workforce of 501-1000 employees. Its website is http://www.sumologic.com.