Applied AI ML Lead

JPMorgan Chase JPMorgan Chase · Banking · Bengaluru, Karnataka, India · Corporate Sector

Lead role in designing, building, and productionizing LLM-powered surveillance systems for regulatory compliance and risk detection within a fintech environment. Focuses on integrating NLP, transformer architectures, and real-time inference with a strong emphasis on engineering discipline and scalability.

What you'd actually do

  1. Design LLM powered features such as risk detection, alert explanation, conversation summarization, reviewer assisted co-pilots
  2. Implement explainability techniques (SHAP, LIME, attention visualization) ensuring model outputs are traceable, versioned and reproducible
  3. Optimize inference latency and token efficiency for production environments
  4. Implement RAG and LLM based risk analysis pipelines processing data at web scale
  5. Bake in augmentation mechanisms leveraging legacy regular expressions for filtering and optimization

Skills

Required

  • 8+ years experience in cloud based applications with 4+ years of experience as an MLE
  • Strong foundation in Information Retrieval, Natural Language Processing and JVM based languages- Python/Kotlin, Java
  • Experience integrating models into cloud scale, microservices based architectures
  • Experience with one or more ML frameworks - Pytorch, Tensorflow, SciKit, NeMo, Huggingface Transformers
  • Hands-on experience with AWS services such as SageMaker, ECS, Lambda functions, Bedrock
  • Experience/Exposure to SQL, NoSQL and messaging stacks
  • Excellent verbal & written communication skills and bias for action and ownership
  • Good understanding of data engineering concepts, distributed systems, and scalable architectures
  • Experience working with NLP, LLMs, embeddings, RAG, or GenAI applications
  • Operational experience in supporting an enterprise grade ML application in production

Nice to have

  • Knowledge of Databricks is nice to have
  • Experience with any of the MLOps frameworks such MLflow, Kubeflow
  • Experience in surveillance, fraud detection, fintech or risk systems is a strong plus
  • Experience building production-grade ML pipelines and APIs
  • Familiarity with vector databases, model serving, and inference optimization is a plus

What the JD emphasized

  • production grade engineering discipline
  • near real-time inference systems
  • regulatory explainability and auditability
  • operational experience in supporting an enterprise grade ML application in production

Other signals

  • LLM integration
  • production grade engineering
  • scalable ML systems
  • near real-time inference