Back to jobs
AS
Hiring companyAU Small Finance Bank Limited

Active listing

AI Engineer

Mumbai, Maharashtra, India Full-time On-site

About the role

AI Engineering team at AU Small Finance Bank builds scalable multi-modal LLM agents and AI solutions for banking services. The role designs, develops, and deploys LLM-based agents, RAG pipelines, and model monitoring, collaborating with data scientists and engineering. Mumbai office, on-site, focusing on rapid AI innovation in finance.

What you’ll do

  • Develop and deploy multi-modal LLM agents using frameworks like LangGraph and Autogen
  • Research and implement RAG, semantic search, and knowledge representation solutions
  • Build and maintain data pipelines, datasets, and model versioning systems
  • Automate model deployment, monitoring, and retraining processes
  • Collaborate with data scientists to optimize ML code and ensure governance

What you’ll bring

  • Bachelor's degree in Computer Science or related field
  • 4-8 years MLOps or DevOps experience
  • Proficient in Python and deep learning frameworks (PyTorch, JAX, TensorFlow)
  • Strong knowledge of machine learning concepts
  • Experience with cloud platforms (AWS, Azure, GCP) and IaC tools like Terraform

Skills

PythonLLMMLOpsCloud PlatformsData PipelinesModel DeploymentFastAPIDocker