AI Lead Software Engineer - Machine Learning jobs in United States
cer-icon
Apply on Employer Site
company-logo

JPMorganChase · 2 days ago

AI Lead Software Engineer - Machine Learning

JPMorganChase, one of the oldest financial institutions, offers innovative financial solutions to millions of consumers and businesses. As a Lead Software Engineer within the Data Products Team, you will collaborate with data scientists and engineering teams to enhance and deliver technology products, focusing on the integration and deployment of machine learning models.

Asset ManagementBankingFinancial Services
check
Growth Opportunities
check
H1B Sponsor Likelynote

Responsibilities

Collaborate with data scientists to facilitate training, fine-tuning, and deployment of ML models, including foundational and generative models
Integrate trained models into production applications (e.g., anomaly detection, automated reporting, agentic AI workflows)
Develop APIs, microservices, and user interfaces to expose model capabilities to business users and other systems
Design and implement prompt engineering strategies and agentic architectures for autonomous AI workflows
Monitor, troubleshoot, and optimize model performance, scalability, and reliability in production environments
Act as a technical liaison between data science, engineering, and product teams to ensure seamless integration and delivery
Document processes, workflows, and best practices for model deployment and application integration

Qualification

PythonML frameworksCloud platformsAPIs/microservicesPrompt engineeringContainerizationCommunicationCollaboration skills

Required

Formal training or certification on software engineering concepts and 5+ years applied experience
Proficiency in Python and experience building APIs/microservices
Experience with ML frameworks (e.g., PyTorch, TensorFlow, Hugging Face) and foundational models (LLMs, generative AI)
Familiarity with prompt engineering and agentic workflows
Strong understanding of cloud platforms (AWS, GCP, Azure) and MLOps practices
Excellent communication and collaboration skills

Preferred

Experience with anomaly detection, automated reporting, or narrative generation systems
Exposure to vector databases, retrieval-augmented generation (RAG), or semantic search
Experience with containerization (Docker, Kubernetes) and CI/CD pipelines
Knowledge of security and compliance in AI/ML deployments
Experience with Databricks ML Ops
Familiarity with regression/classification models and their integration into production systems

Benefits

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

Company

JPMorganChase

company-logo
With a history tracing its roots to 1799 in New York City, JPMorganChase is one of the world's oldest, largest, and best-known financial institutions—carrying forth the innovative spirit of our heritage firms in global operations across 100 markets.

H1B Sponsorship

JPMorganChase has a track record of offering H1B sponsorships. Please note that this does not guarantee sponsorship for this specific role. Below presents additional info for your reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (3471)
2024 (3469)
2023 (3395)
2022 (3594)
2021 (2515)
2020 (2495)

Funding

Current Stage
Public Company
Total Funding
unknown
1998-02-01IPO

Leadership Team

leader-logo
Allison Beer
CEO of Card Services and Connected Commerce
linkedin
leader-logo
Dan Mendelson
CEO, Morgan Health
linkedin
Company data provided by crunchbase