Senior Machine Learning Engineer

Sumo Logic Sumo Logic · Enterprise · India · Software Engineering

Senior Machine Learning Engineer to build intelligence behind agentic AI systems that reason over massive, heterogeneous log data, combining machine learning, prompt engineering, and rigorous evaluation. Focus on LLM infrastructure and AI observability.

What you'd actually do

  1. Design, implement, and optimize agentic AI components, including context engineering, memory management, and prompts.
  2. Collaborate cross-functionally with product, data, and infrastructure teams to deliver end-to-end AI-powered insights.
  3. Operate autonomously in a fast-paced, ambiguous environment - defining scope, setting milestones, and driving outcomes.
  4. Ensure reliability, performance, and observability of deployed agents through rigorous testing and continuous improvement.
  5. Maintain a strong bias for action—delivering incremental, well-tested improvements that directly enhance customer experience.

Skills

Required

  • Python
  • ML/data libraries (NumPy, pandas, scikit-learn)
  • LLM core concepts
  • prompt design
  • agentic design patterns
  • machine learning fundamentals
  • data pipelines
  • model evaluation

Nice to have

  • JVM languages
  • building and deploying AI agents or LLM applications in production
  • modern agentic AI frameworks (e.g., LangGraph, LangChain, CrewAI)
  • ML infrastructure and tooling (PyTorch, MLflow, Airflow, Docker, AWS)
  • LLM Ops - infrastructure optimization, observability, latency, and cost monitoring

What the JD emphasized

  • agentic AI systems
  • agentic design patterns
  • building and deploying AI agents
  • agentic AI frameworks

Other signals

  • building agentic AI systems
  • reason over massive, heterogeneous log data
  • LLM infrastructure
  • AI observability