All roles at KEY

Applied AI Engineer

BengaluruFull-time4–7 yrsPosted 1 week ago

Mid-level data science role at KEY in Bengaluru. Needs 4+ years and an M.Sc, B.Tech or B.E..

About the role

Were looking for an Applied AI Engineer to take AI and ML models from concept to production. You'll build scalable, reliable systems using Python, cloud services, and modern MLOps practices, and own end-to-end delivery of AI solutions.

What you'll do

  • End-to-end ownership of production ML systems from prototype to live services
  • Implement, fine-tune, and optimize models, including LLMs for search and recommendation, for performance, latency, and cost
  • Design, build, and maintain data and inference pipelines using MLOps principles for CI/CD, monitoring, and automated retraining
  • Collaborate with backend and frontend to integrate AI into product experiences
  • Design and maintain well-documented, scalable APIs and services to expose AI capabilities

What you need

  • Machine learning
  • Python
  • Scikit-learn
  • MLOps
  • LLMs
  • AWS

Good to have

  • PyTorch
  • TensorFlow
  • Airflow
  • CI/CD
  • GCP
  • Node.js

More roles like this at KEY

Apply on KEY's site