Python Software Engineer III - AI/ML

Job Description

We have an exciting and rewarding opportunity for you to take your software engineering career to the next level.

As a Software Engineer III at JPMorgan Chase within the Commercial and Investment Bank organization, you will be a seasoned member of an agile team, tasked with designing and delivering trusted, market-leading technology products that are secure, stable, and scalable. Your responsibilities include implementing critical technology solutions across multiple technical areas, supporting various business functions to achieve the firm's business objectives.

Job responsibilities

• Engages in platform engineering to enable the Gen AI platform, develop Gen AI use cases, LLM fine-tuning, and multi-agent orchestration
• Builds machine learning pipelines using Generative AI models on AWS and implements best practices in software engineering, MLOps, and data governance
• Maintains consistent code check-ins every sprint for continuous integration and development
• Ensures AI/ML solutions comply with data privacy and security regulations
• Shares technical concepts and solutions across all organizational levels while proactively working to understand key business problems and processes
• Executes tasks throughout the model development process, including data wrangling, analysis, model training, testing, and selection
• Generates and presents structured insights from data analysis and modeling exercises tailored to the audience
• Conducts ad-hoc and periodic analysis as required by business stakeholders, the model risk function, and other groups
• Collaborates with engineers to build and deploy solutions for business requirements
• Identifies and addresses AI/ML/LLM/GenAI challenges, implement optimizations, and fine-tune models for optimal performance in Natural Language Processing (NLP) applications
• Works with the OpenAI API, including prompt engineering, vector databases, and AI agents

Required qualifications, capabilities, and skills

• Formal training or certification on software engineering concepts and 3+ years applied experience
• Proficient coding experience in Python and relevant libraries (TensorFlow, PyTorch, scikit-learn)
• Experience with machine learning frameworks, libraries, and APIs, such as TensorFlow, PyTorch, Scikit-learn, and OpenAI API
• Expertise in data wrangling, including understanding, cleaning, reshaping, and joining complex datasets using Python
• Advanced knowledge in software engineering, AI/ML, MLOps, and data governance
• Expertise in AWS cloud services, including ECS, RDS, DynamoDB, S3, Lambda, SageMaker, and EC2
• Experience with cloud computing platforms (AWS, Azure, or Google Cloud Platform), containerization technologies (Docker and Kubernetes), microservices design, and performance optimization
• Demonstrated experience with Natural Language Processing (NLP), including tokenization, embedding's, sentiment analysis, and basic transformers for text-heavy datasets
• Experience with LLM & Prompt Engineering, including tools such as LangChain, LangGraph, and Retrieval-Augmented Generation (RAG)
• Strong collaboration skills to work effectively with cross-functional teams, communicate complex concepts, and contribute to interdisciplinary projects
• Demonstrated experience in applied AI/ML engineering, with a track record of developing and deploying business-critical machine learning models in production

Preferred qualifications, capabilities, and skills

• Familiarity with Java and React
• Experience with big data frameworks, with a preference for Databricks
• Experience with databases, including SQL (Oracle, Aurora), and Vector DB
• Familiarity with version control systems such as Bitbucket and GitHub
• Proficient knowledge of graph analytics and neural networks
• Experience working with engineering teams to operationalize machine learning models

About Us

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

About the Team

J.P. Morgan's Commercial & Investment Bank is a global leader across banking, markets, securities services and payments. Corporations, governments and institutions throughout the world entrust us with their business in more than 100 countries. The Commercial & Investment Bank provides strategic advice, raises capital, manages risk and extends liquidity in markets around the world.