Machine Learning Engineer - Document Digitization (LLMs)-Senior Associate

Aumni; Jersey City (NJ); On-site; USD 120,000 - 180,000; Full time

Benefits offered by this job

Competitive compensation

Health benefits

401(k)

Paid time off

Job summary

JPMorgan Chase & Co. is seeking a senior Machine Learning Engineer – Document Digitization to design and deploy secure AI/ML solutions for processing diverse documents. You will lead end-to-end ML workflows, apply generative AI, and build scalable pipelines in production.

You will mentor engineers, collaborate across teams, and ensure governance, security, and compliance while delivering measurable business impact.

Qualifications

• Bachelor's or Master's degree in Computer Science, Data Science, ML or related field.
• Proficiency in Python; familiarity with Java and front-end technologies (React.js, AngularJS).
• Hands-on AI/ML model development, deployment, and MLOps in production.
• Experience with TensorFlow, PyTorch, scikit-learn, PyTorch Lightning.
• Knowledge of generative AI techniques and LLMs.
• Experience with AWS SageMaker, Bedrock; Docker, Kubernetes; Terraform.
• Familiarity with NoSQL and search tech (MongoDB Atlas, Elasticsearch, OpenSearch, Neo4j).
• Experience with autonomous code agents and agentic development workflows.
• Understanding SDLC, CI/CD, resiliency and security in production.

Responsibilities

• Design, develop, and integrate AI-powered document digitization solutions to extract information and insights.
• Manage end-to-end AI/ML lifecycle: training, validation, deployment, monitoring, and improvement.
• Use generative AI and LLMs to automate document workflows.
• Build scalable digitization pipelines with Python, AI frameworks, and cloud tech.
• Provision and manage cloud resources with Terraform and AWS services (SageMaker, Bedrock).
• Ensure scalability, security, and governance of AI/ML solutions.
• Collaborate with cross-functional teams to modernize legacy processes with AI/LLMs.
• Develop dashboards to monitor digitization accuracy and efficiency.
• Mentor junior engineers and promote best practices in AI/ML and software engineering.
• Conduct model validation and continuous improvement with human-in-the-loop.

Skills

Python

TensorFlow/PyTorch

MLOps

AWS SageMaker

Terraform

Docker/Kubernetes

LLMs

Java

React/Angular

Education

Bachelor's or Master's in Computer Science/Data Science/ML

Tools

SageMaker

Bedrock

Docker

Kubernetes

Terraform

MongoDB Atlas

Elasticsearch

OpenSearch

Neo4j

Job description

Are you passionate about leveraging advanced technology to solve complex business challenges? As an applied AI/ML, you will have the opportunity to shape the future of document management through cutting-edge AI and machine learning. Join a collaborative team where your expertise will drive impactful solutions and strategic outcomes. This role offers a platform to innovate, lead, and make a difference across the organization.

As a Machine Learning Engineer - Document Digitization (LLMs) - Senior Associate within the Global Technology team, you will design, develop, and deploy secure, scalable, and innovative technology products that transform how documents are processed and managed. You will use advanced AI and machine learning to extract, analyze, and manage information, driving strategic business outcomes. You will collaborate with cross-functional teams, mentor others, and continuously seek opportunities for improvement and innovation.

Job responsibilities

• Design, develop, and integrate AI-powered document digitization solutions, focusing on extracting information and insights from diverse document types.
• Manage the end-to-end AI/ML lifecycle: model training, validation, deployment, monitoring, and continuous improvement in production environments.
• Employ generative AI and large language models (LLMs) to automate and optimize document workflows.
• Build and maintain scalable document digitization pipelines using Python, AI frameworks, and cloud technologies.
• Provision and manage cloud resources using infrastructure as code tools (Terraform) and AWS services (SageMaker, Bedrock).
• Ensure scalability, reliability, security, and compliance of AI/ML solutions, adhering to best practices and governance standards.
• Collaborate with cross-functional teams to reimagine legacy document processing systems using generative AI and LLMs.
• Develop and maintain dashboards and reporting tools to monitor digitization accuracy, workflow efficiency, and business impact.
• Mentor junior engineers and promote best practices in AI/ML, software engineering, and testing.
• Conduct model validation, human-in-the-loop review, and implement continuous improvement strategies for digitization accuracy.
• Contribute to communities of practice and explore new and emerging technologies.

Required qualifications, capabilities, and skills

• Bachelor's or Master's degree in Computer Science, Data Science, Machine Learning, or a related field, with relevant industry experience.
• Proficiency in Python; familiarity with Java and front-end technologies (React.js, AngularJS).
• Hands-on experience with AI/ML model development, deployment, and MLOps practices in production environments.
• Experience with machine learning frameworks (TensorFlow, PyTorch, scikit-learn, PyTorch Lightning).
• Knowledge of generative AI techniques (GANs, VAEs, transformers, diffusion models) and LLMs.
• Experience with AWS services (SageMaker, Bedrock), containerization (Docker, Kubernetes, Amazon Elastic Kubernetes Service (EKS)), and infrastructure as code (Terraform).
• Familiarity with NoSQL and search technologies (MongoDB Atlas, Elasticsearch, OpenSearch, Neo4j).
• Experience using autonomous code agents and agentic development workflows for code generation and refactoring.
• Understanding of the software development lifecycle (SDLC), continuous integration/continuous delivery (CI/CD), resiliency, and security practices.
• Experience deploying and maintaining AI/ML solutions in large-scale production environments; effective problem-solving, communication, and collaboration skills.

Preferred qualifications, capabilities, and skills

• Experience in financial services, including investment banking or credit risk operations.
• Experience with agentic AI frameworks, prompt optimization, and fine-tuning small language models (SLMs).
• Familiarity with distributed computing, data sharing, and distributed data parallel (DDP) training.
• Experience in design reviews, code reviews, and mentoring teams.
• AWS Generative AI Developer - Professional certification or equivalent.

JPMorganChase, one of the oldest financial institutions, offers innovative financial solutions to millions of consumers, small businesses and many of the world’s most prominent corporate, institutional and government clients under the J.P. Morgan and Chase brands. Our history spans over 200 years and today we are a leader in investment banking, consumer and small business banking, commercial banking, financial transaction processing and asset management.

We offer a competitive total rewards package including base salary determined based on the role, experience, skill set and location. Those in eligible roles may receive commission-based pay and/or discretionary incentive compensation, paid in the form of cash and/or forfeitable equity, awarded in recognition of individual achievements and contributions. We also offer a range of benefits and programs to meet employee needs, based on eligibility. These benefits include comprehensive health care coverage, on-site health and wellness centers, a retirement savings plan, backup childcare, tuition reimbursement, mental health support, financial coaching and more. Additional details about total compensation and benefits will be provided during the hiring process.

We recognize that our people are our strength and the diverse talents they bring to our global workforce are directly linked to our success. We are an equal opportunity employer and place a high value on diversity and inclusion at our company. We do not discriminate on the basis of any protected attribute, including race, religion, color, national origin, gender, sexual orientation, gender identity, gender expression, age, marital or veteran status, pregnancy or disability, or any other basis protected under applicable law. We also make reasonable accommodations for applicants' and employees' religious practices and beliefs, as well as mental health or physical disability needs. Visit our FAQs for more information about requesting an accommodation.

JPMorgan Chase & Co. is an Equal Opportunity Employer, including Disability/Veterans

Our professionals in our Corporate Functions cover a diverse range of areas from finance and risk to human resources and marketing. Our corporate teams are an essential part of our company, ensuring that we’re setting our businesses, clients, customers and employees up for success. Machine learning engineer to build secure generative AI and LLM document digitization pipelines on AWS, leading end-to-end MLOps.