All roles at UST

Lead III - ML Engineering

Bengaluru, Hyderabad, Pune, ThiruvananthapuramFull-time9–12 yrsPosted 1 week ago

Leadership data science role at UST in Bengaluru, Hyderabad and 2 more cities. Needs 9–12 years.

About the role

Develop, test, and optimize LLM-powered applications, including prompt engineering, context management, orchestration logic, memory frameworks, and agent-tool interactions to improve accuracy, performance, and user experience. Collaborate with architecture, cloud, and DevOps teams to implement secure, scalable, and observable AI solutions following software engineering best practices, CI/CD standards, and enterprise governance requirements.

What you'll do

  • Lead AI Developer Design, develop, and deploy Generative AI and Agentic AI applications on AWS
  • Build intelligent agents and multi-agent workflows using LangGraph, LangChain, Python, and Retrieval-Augmented Generation (RAG) patterns
  • Develop, test, and optimize LLM-powered applications, including prompt engineering, context management, orchestration logic, memory frameworks
  • Collaborate with architecture, cloud, and DevOps teams to implement secure, scalable, and observable AI solutions following software engineering best practices

What you need

  • Machine learning
  • Python
  • LLMs
  • AWS
  • Prompt engineering
  • CI/CD

More roles like this at UST

Apply on UST's site