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

JPMorgan Chase & Co. • Jersey City, NJ

14h • $137,750 - $195,000

About The Position

As a Senior Associate on the Global Technology team, the Machine Learning Engineer will design, develop, and deploy secure, scalable technology products that transform document processing and management. The role applies AI and machine learning to extract, analyze, and manage information from diverse document types, supporting strategic business outcomes. It includes collaboration with cross-functional teams, mentoring colleagues, and pursuing continuous improvement and innovation.

Requirements

• 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 such as React.js and AngularJS.
• Hands-on experience developing and deploying AI/ML models and applying MLOps practices in production environments.
• Experience with machine learning frameworks including TensorFlow, PyTorch, scikit-learn, or PyTorch Lightning.
• Knowledge of generative AI techniques, including GANs, VAEs, transformers, and diffusion models, and of LLMs.
• Experience with AWS services such as SageMaker and Bedrock; Docker, Kubernetes, and Amazon EKS; and Terraform.
• Familiarity with NoSQL and search technologies such as MongoDB Atlas, Elasticsearch, OpenSearch, and Neo4j.
• Experience using autonomous code agents and agentic development workflows for code generation and refactoring.
• Understanding of the software development lifecycle, 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.

Nice To Haves

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

Responsibilities

• Design, develop, and integrate AI-powered document digitization solutions to extract information and insights from diverse document types.
• Manage the end-to-end AI/ML lifecycle, including model training, validation, deployment, monitoring, and continuous improvement in production.
• Use generative AI and large language models 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 Terraform and AWS services, including SageMaker and Bedrock.
• Ensure AI/ML solutions are scalable, reliable, secure, and compliant with 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 and human-in-the-loop review, and implement continuous improvement strategies for digitization accuracy.
• Contribute to communities of practice and explore emerging technologies.

Benefits

• Competitive total rewards package, including base salary determined by role, experience, skill set, and location.
• Eligible roles may receive commission-based pay and/or discretionary incentive compensation in cash and/or forfeitable equity.
• Comprehensive health care coverage.
• On-site health and wellness centers.
• Retirement savings plan.
• Backup childcare.
• Tuition reimbursement.
• Mental health support.
• Financial coaching.

What This Job Offers

• Job Type: Full-time
• Career Level: Senior