All jobs
Axis Bank logo

EAIG:Head - Data Science

Axis Bank · Posted today

  • Bengaluru East, Karnataka, India (On-site)
  • Full-time
  • 6+ yrs

About the role

Role Description

About the Role

Build, tune, evaluate and serve our own small language and ML models in-house. Role will own the science end to end, base model to production, with the evals and model cards to prove it.

Key Responsibilities

  • Train, fine-tune and distil models—genAI (LoRA/PEFT, RLHF, prompt-based) and classical (transfer learning)—for live product use cases.
  • Design evaluation, model tracking and model cards: rubrics, LM-as-judge, regression, drift.
  • Own the "tune vs. swap the base model" call; partner with platform to deploy, serve and maintain foundry models.
  • Set method and mentor the DS/ML team.

Role Proficiency

  • Fine-tuning/PEFT, transfer learning, RLHF, prompt tuning; evaluation and experiment tracking.
  • Strong probability and statistical inference—distributions, Bayesian methods, significance, calibration, experiment/A-B design; and sound handling of skewed/imbalanced data (resampling, class weighting).
  • Some MLOps— deploy/serve/monitor (SageMaker, MLflow, Docker; vLLM/Triton a plus).
  • MS/PhD in CS/Stats preferred. Fluent in Python/Jupyter, PyTorch, Hugging Face; SQL/Postgres (pgvector), a vector store, a warehouse (Snowflake/BigQuery); Spark or Ray; Airflow/dbt.
  • Shipped many models in a large org; strong written and verbal communication skills (tech blogs, publications, talks at meetups/conferences).

Qualifications

  • Post-graduate degree (MBA, MTech, LLM, or equivalent) from a top-tier institution.
  • 6+ yrs building ML/AI in production, deep in model training—genAI or classical (at least one; both preferred)—with 3+ yrs leading or tech-leading.

Good To Have

  • GenAI product experience (e.g. enterprise RAG bot); text, voice and vision (VLM) models; startup / fast-paced background

Skills

Fine-tuning/PEFTTransfer learningRLHFPrompt tuningEvaluation and experiment trackingProbability and statistical inferenceHandling skewed/imbalanced dataMLOps (SageMaker, MLflow, Docker)Python/JupyterPyTorchHugging Face
EAIG:Head - Data Science at Axis Bank, Bengaluru East, Karnataka, India (On-site) | Quark9