Lead Software Engineer - AI/ML Deep Learning & GPU ML Serving

JPMorgan Chase • Palo Alto, CA • 3h ago

Be an integral part of an agile team that's constantly pushing the envelope to enhance, build, and deliver top-notch technology products.

As a Lead Software Engineer at JPMorgan Chase within the Commercial and Investment Banking team, you will play a pivotal role in an agile team, enhancing and delivering secure, stable, and scalable technology products. As a core technical contributor, you will drive critical technology solutions across multiple technical areas, supporting the firm's business objectives.

Job Responsibilities

Lead the design, development, and troubleshooting of software solutions, applying innovative approaches to complex technical challenges.

Write secure, high-quality production code and maintain algorithms integrated with firm systems.

Produce architecture and design artifacts for advanced applications, ensuring compliance with design constraints.

Analyze and visualize large, diverse data sets to improve software applications and systems.

Identify and resolve hidden issues and patterns in data to enhance code quality and system architecture.

Collaborate with software engineering communities to explore and adopt emerging technologies.

Guide system design and architecture discussions, focusing on reliability and scalability.

Optimize deep learning models for production inference, including quantization and batching.

Deploy and manage GPU workloads in Kubernetes environments.

Build scalable, low-latency systems using web services and APIs.

Partner with product and program management teams to deliver business-driven solutions.

Required qualifications, capabilities, and skills

Formal training or certification on software engineering concepts and 5+ years applied experience

Professional software development experience, with emphasis on ML systems.

Strong proficiency in Python and experience with ML frameworks (TensorFlow, PyTorch, or similar).

Experience with cloud technologies (Docker, Kubernetes, EKS) and public clouds (AWS, GCP).

Hands-on experience with ML model serving frameworks (TorchServe, TensorFlow Serving, Triton Inference Server).

Experience deploying and managing GPU workloads in Kubernetes.

Familiarity with scalable, low-latency systems based on web services and APIs.

Experience with NoSQL databases (Cassandra or equivalent) for high-throughput data access.

Understanding of GPU resource management and cost optimization.

Experience with modern microservices architecture.

Ability to lead the design of large-scale systems and evaluate tradeoffs.

Preferred qualifications, capabilities, and skills

MS/PhD in Computer Science, Machine Learning, or a related field.

Proficiency in Java, Python, Scala, or C++.

Experience with graph neural networks and graph processing frameworks (DGL, PyTorch Geometric, NetworkX).

Knowledge of GPU programming (CUDA) and performance optimization.

Experience with model monitoring, A/B testing, and ML observability tools.

Familiarity with MLOps tools and practices (MLflow, Kubeflow, SageMaker).

Experience serving large-scale models and optimizing for performance.

FEDERAL DEPOSIT INSURANCE ACT:

This position is subject to Section 19 of the Federal Deposit Insurance Act. As such, an employment offer for this position is contingent on JPMorgan Chase's review of criminal conviction history, including pretrial diversions or program entries.

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.

Coherent • Sunnyvale, CA • 3h ago

IT & Engineering

Job Description Primary Duties & Responsibilities Execute and maintain manufacturing processes in accordance with established SOPs to ensure consistent testing, integration, and shipment of OCS systems. Support daily...

3h ago

Go to Apply

Staff, Software Engineer, Information Security, Non-Human Identity / Agentic Identity

Walmart • Sunnyvale, CA • 4h ago

IT & Engineering

Position Summary... We are seeking a Staff Software Engineer, Information Security to design, build, and scale identity, authentication, and authorization capabilities for agentic AI ecosystems. This role is focused on N...

4h ago

Go to Apply

Robot Operator/Data Collector

Aerotek • San Jose, CA • 5h ago

IT & Engineering

Hiring Robot Operators/Data Collectors. No experience required and training will be provided. As a Data Collector, you will wear sensor equipment to guide a robot through human-like motions. You will be responsible for c...

5h ago

Go to Apply

Subscribe to get new jobs sent directly to you
(no charge, cancel anytime)

This site is protected by reCAPTCHA and the Google Privacy Policy and Terms of Service apply.