Lead Software Engineer - Machine Learning Platform
Palo Alto, California, United States · On-site
$152k–$215k/yr
Senior$1.6B raised
We have an opportunity to impact your career and provide an adventure where you can push the limits of what's possible. As a Lead Software Engineer at JPMorganChase within AI/ML Data Platforms, you are an integral part o…
Who We Are The world is moving towards instant digital payments and TabaPay is leading the way. We help thousands of Fintechs in the US and Canada instantly move money in and out of accounts and we are actively expanding…
Skills: DevOps, Site Reliability Engineering, Python, Java, Go
About Summit: Summit Therapeutics Inc. is a biopharmaceutical oncology company with a mission focused on improving quality of life, increasing potential duration of life, and resolving serious unmet medical needs. At Sum…
About the Role As a seasonal Tax Associate at H&R Block, you’ll begin a career in tax and client services—no prior tax experience required. With training and support from day one, you’ll work directly w…
Skills: Tax preparation, Client services, Communication, Problem-solving, Customer service
H&R Block
Tax Associate
Palo Alto, California, United States · On-site
Entry level
About the Role As a seasonal Tax Associate at H&R Block, you’ll begin a career in tax and client services—no prior tax experience required. With training and support from day one, you’ll work directly w…
Skills: Tax preparation, Client services, Communication, Problem-solving, Customer service
Our Purpose At Fiddler, we understand the implications of AI and the impact that it has on human lives. Our company was born with the mission of building trust into AI. The rise of Generative AI and Agents has unlocked g…
Skills: Enterprise sales, AI observability, Machine learning, Big data, Strategic account management
About the Role We're seeking a Founding Go-To-Market (GTM) expert who will play a foundational role in building our sales and account management capabilities at Pika. As the first member of our GTM team, you will take ow…
Skills: Sales, Account management, Go-to-market strategy, Customer acquisition, Business development
Uniphore is one of the largest B2B AI-native companies—decades-proven, built-for-scale and designed for the enterprise. The company drives business outcomes, across multiple industry verticals, and enables the largest gl…
Skills: AI System Design, Generative AI, Machine Learning, Model Optimization, Agentic Workflows
Business Unit What the Role EntailsAbout the Company: Tencent is a leading global technology company focused on connecting people and developing innovative products and services that improve the quality of life of people…
About The RoleThe Visual Merchandiser is a key position in our Retail stores at The RealReal. As a Visual Merchandiser you are responsible for the overall aesthetic and functionality of the store's visual and merchandisi…
The Mission Our mission is to deliver everyone a sustainable income. Right now 50,000 users make money on Whop every month, and this year users will earn over $5B. We will not stop until 1B people every month are making …
Skills: Problem-solving, Systems engineering, Data analysis, Software development
The Amazon Search team creates customer-focused search solutions and technologies. Whenever a customer visits an Amazon site worldwide and types in a query or browses through product categories, Amazon Product Search ser…
Clinical Lead & Product Manager, Care Delivery Onsite in Palo Alto, CA: Hybrid Tues/Wed/Thurs Transform healthcare with us. At Qualified Health, we're redefining what's possible with Generative AI in healthcare. Our infr…
Skills: Clinical leadership, Product management, Generative AI, Care delivery workflows, Population health
Transform healthcare with us. At Qualified Health, we’re redefining what’s possible with Generative AI in healthcare. Our infrastructure provides the guardrails for safe AI governance, healthcare-specific agent creation,…
Why Join GEICO? At GEICO, we offer a rewarding career where your ambitions are met with endless possibilities. Every day we honor our iconic brand by offering quality coverage to millions of customers and being there whe…
Business Unit What the Role Entails Own end-to-end operations of Tencent's overseas Elastic IP and load balancing gateway platform, covering user support ticket resolution, software deployment, on-call rotations, and cri…
Business Unit What the Role Entails Hunt and generate new business opportunities by identifying and targeting key accounts Build and maintain strong relationships with key customers, understanding their needs and providi…
IVD Documentation Scientist, Manufacturing and Medical Device (Contract)
Palo Alto, California, United States · Hybrid
$62/hr–$80/hr
Senior$332M raised
About Us DELFI Diagnostics, Inc. (DELFI Diagnostics) is developing next-generation, blood-based tests that are reliable, accessible and deliver a new way to help detect cancer. Employing advanced machine-learning methods…
Skills: GMP, ISO 13485, IVD, Medical Device Documentation, Batch Records
About Us Rivian and Volkswagen Group Technologies is a joint venture between two industry leaders with a clear vision for automotive’s next chapter. From operating systems to zonal controllers to cloud and connectivity s…
Skills: Systems integration, Embedded systems, Vehicle diagnostics, CAN, LIN
Lead Software Engineer - Machine Learning Platform
Sign up with Clera and we'll reach out the moment a role actually fits you — no more spraying applications into the void.
$152k–$215k/yr
Full-time
Health care coverage, Retirement savings plan, Backup childcare, Tuition reimbursement, Mental health support, Financial coaching
Posted 33d ago
~40 hrs/week
Responsibilities
The Lead Software Engineer will design, build, and maintain scalable ML training platforms on AWS while optimizing GPU workloads and infrastructure. They will also drive team adoption of AI-assisted engineering practices and ensure robust observability and governance across ML pipelines.
Requirements
Candidates must have 5+ years of software engineering experience with demonstrated expertise in running ML training in cloud environments. Strong proficiency in Python, deep learning frameworks, and Kubernetes is required, along with the ability to profile and optimize compute-intensive workloads.
Full job description
We have an opportunity to impact your career and provide an adventure where you can push the limits of what's possible.
As a Lead Software Engineer at JPMorganChase within AI/ML Data Platforms, you are an integral part of an agile team that works to build and operate scalable, reliable ML training systems and pipelines on AWS and other cloud platforms. You will productionize training workloads (often GPU-based), improve performance and cost efficiency, and enable repeatable, well-governed training across environments.
Job Responsibilities
Design, build, and maintain end-to-end ML training platform.
Run and optimize GPU training workloads (single-node and distributed), improving throughput, utilization and reproducibility.
Build and operate training infrastructure on Kubernetes (e.g., EKS and other manage Kubernetes platforms), including resource management and workload troubleshooting.
Enable Gen AI/LLM training and fine-tuning workflows (e.g., supervised fine-tuning), including evaluation harnesses, artifact/version governance, and scalable GPU execution patterns aligned to enterprise controls.
Implement observability for training systems: metrics, logs, dashboards, alerting, and operational runbooks.
Partner with data engineering and platform teams to define interfaces, standards, and guardrails (security, access, cost controls)
Improve developer experience for training: standardized containers, CI/CD, templates, documentation, and self-service workflow
Drives team adoption of enterprise-authorized AI-assisted engineering practices within the work environment to improve code quality, delivery speed, and operational outcomes (e.g., AI-assisted code review/refactoring, test strategy acceleration, incident/root-cause analysis support), while establishing consistent validation standards (secure coding, peer review, automated testing) and promoting reuse of effective patterns across the team.
Applies knowledge of tools within the Software Development Life Cycle toolchain, including enterprise-authorized AI-assisted development and automation capabilities, to improve the value realized by automation.
Required qualifications, capabilities, and skills
Formal training or certification on software engineering concepts and 5+ years applied experience.
Demonstrated experience running ML training in cloud environments and debugging issues across infrastructure & code.
Experience building automation/CI for ML codebases (build, test, release, deployment/promotion workflows).
Hands on experience with deep learning training workflows and at least one major framework (eg., PyTorch or TensorFlow).
Understanding of training performance and stability: data loading bottlenecks, mixed precision, checkpointing, reproducibility, and evaluation methodology.
Experience with distributed training and related concepts (e.g., DDP/FSDP/DeepSpeed concepts, collective communication basics, scaling and bottleneck analysis).
Ability to profile and optimize training systems (CPU/GPU utilization, memory, I/O throughput, networking, scheduling).
Experience with Kubernetes fundamentals for running compute-intensive workloads and AWS (eg., EKS/ECR, S3, IAM, VPC/networking, Cloudwatch, EC2)
Demonstrated experience leading effective use of approved AI-assisted software development tools (e.g., for coding, code review, test acceleration, troubleshooting) with the ability to set team expectations for validating AI outputs for correctness, performance, and security.
Strong understanding of responsible AI use in engineering workflows, including data sensitivity considerations, secure handling of inputs/outputs, and adherence to resiliency and security expectations; experience coaching engineers on safe, compliant adoption within delivery practices.
Preferred qualifications, capabilities, and skills
Experience running training workloads across multiple cloud platforms and managing portability, performance, and governance across environments.
Familiarity with cloud-native networking/storage patterns for high-throughput training and artifact management.
Experience optimizing training input pipelines (sharding, prefetching, caching, format choices such as Parquet/WebDataset) and working with large datasets.
Familiarity with distributed compute frameworks (Spark, Ray, Dask) for feature/dataset generation.
Familiarity with workflow orchestration tools (Airflow-like systems, Argo Workflows-like patterns) and model registry concepts.
Experience optimizing training cost/performance (right-sizing, scheduling policies, interruptible capacity strategies where applicable budge guardrails, quota planning).
Strong observability practice for training systems: metrics/logs/traces, GPU telemetry, dashboards, and alert tuning.
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 JPMorganChase’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.
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.
Related keywords
Machine LearningAWSKubernetesPythonGPUPyTorchTensorFlowEKSCI/CDGen AILLMDistributed TrainingDeep LearningCloud ComputingObservabilitySoftware Development Life Cycle
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.
We serve millions of customers and many of the world’s most prominent corporate, institutional, and government clients daily, managing assets and investments, offering business advice and strategies, and providing innovative banking solutions and services.
Social Media Terms and Conditions: https://bit.ly/JPMCSocialTerms
JPMorgan Chase & Co. is an Equal Opportunity Employer. All qualified applicants will receive consideration for employment without regard to race, color, religion, sex, sexual orientation, gender identity, national origin, disability, or status as a protected veteran.
Offices: 270 Park Avenue, New York, NY 10017-2014, US · 101 Duncan Mill Road, Toronto, ON M3B1Z3, CA · 2200 Ross Ave, Dallas, TX 75201, US · Kadubeesanahalli Circle, Bengaluru, Karnataka 560103, IN · 370 S Cleveland Ave, Westerville, OH 43081, US
Consumer & Community BankingCorporate and Investment BankAsset ManagementPrivate Bankingand Commercial BankingConstructionResidentialFinanceBankingBusiness Development