Applied AI ML Associate Senior

JPMorgan Chase JPMorgan Chase · Banking · Hyderabad, Telangana, India · Commercial & Investment Bank

JPMorgan Chase Payments is seeking an Applied AI/ML Associate Senior to build, train, and debug neural network workflows end-to-end for document and image understanding tasks. The role involves delivering robust ML capabilities under real-world constraints, with a focus on sensitive data, and contributing to production integration and monitoring.

What you'd actually do

  1. Build end-to-end deep learning workflows, including data preprocessing, dataset construction, train/validation/test splits, and leakage prevention checks.
  2. Implement and debug neural network models (e.g., Transformers, CNNs, LSTMs) for document understanding tasks such as classification, ranking, entity extraction, and language understanding.
  3. Create rigorous evaluation harnesses: offline metrics, calibration/thresholding, robustness testing, slice/segmentation analysis, and systematic error analysis.
  4. Optimize models for deployment constraints using techniques such as parameter-efficient fine-tuning, distillation, quantization, batching strategies, and latency profiling.
  5. Contribute to production integration: model packaging, validation tests, inference pipelines (batch and real-time), monitoring/alerting, and rollback procedures.

Skills

Required

  • PyTorch or TensorFlow
  • deep learning frameworks
  • modern neural architectures
  • model development fundamentals
  • preprocessing
  • training loops
  • hyperparameter tuning
  • evaluation design
  • failure-mode analysis
  • deploying or integrating models into services/pipelines on AWS
  • operating them reliably at scale
  • communication skills
  • articulating tradeoffs

Nice to have

  • document AI
  • OCR pipelines
  • layout-aware NLP
  • document classification/ranking
  • entity extraction
  • form understanding
  • fine-tuning approaches
  • model optimization techniques
  • distillation
  • quantization-aware approaches
  • compression
  • ML platform/MLOps practices
  • data validation
  • model registries
  • CI/CD for ML
  • observability/monitoring
  • governance workflows
  • Docker/Kubernetes
  • modern data platforms
  • Databricks
  • Snowflake
  • streaming or near-real-time architectures
  • feature generation
  • point-in-time correctness
  • leakage prevention
  • SQL
  • distributed processing
  • Spark/PySpark

What the JD emphasized

  • implementing, training, and debugging neural network workflows end-to-end
  • deliver robust ML capabilities under real-world constraints (latency, scale, reliability, explainability, and governance)
  • special care given to sensitive data
  • training and shipping neural network-based models into production
  • operating them reliably at scale

Other signals

  • end-to-end deep learning workflows
  • document understanding tasks
  • production inference
  • real-world constraints
  • sensitive data