Lead Software Engineer - Machine Learning Engineer

Job Description
Be an integral part of an agile team that's constantly pushing the envelope to enhance, build, and deliver top-notch technology products.
Are you ready to make a significant impact in the world of AI and machine learning? At JPMorgan Chase, In this role, you'll be a key player in an agile team dedicated to enhancing, building, and delivering market-leading technology products that are secure, stable, and scalable
As a Senior Lead Software/Machine Learning Engineer, at JPMorgan Chase within the Corporate Sector - AIML Data Platforms Team, you are an integral part of an agile team that works to enhance, build, and deliver trusted market-leading technology products in a secure, stable, and scalable way. Drive significant business impact through your capabilities and contributions and apply deep technical expertise and problem-solving methodologies to tackle a diverse array of challenges that span multiple technologies and applications.
Job responsibilities

• Architects and implements distributed ML experimentation and training platform for firm-wide use.
• Designs, implements, and supports tools and workflows to facilitate machine learning experiments, automated training runs, and production deployments
• Extends machine learning libraries and frameworks to support complex requirements
• Designs thoughtful data scientist experience in delivering AI experience APIs and SDKs for the platform
• Collaborates with infrastructure engineering, product management, and security and compliance teams to deliver tailored, robust solutions

Required qualifications, capabilities, and skills

• Formal training or certification on software engineering concepts and 5+ years applied experience
• Advanced knowledge of architecture, design, and software development processes.
• Deep understanding and hands-on experience with public cloud technologies, especially with AWS, in the context of ML engineering workflows, specifically featurization, experimentation, training, and evaluation
• Expert programming skills in at least Python and experience with ML frameworks and libraries such as TensorFlow, PyTorch, Scikit-Learn, JAX, etc.
• Hands-on experience implementing DevOps practices using tools such as Docker, Jenkins, Spinnaker, and Terraform
• Knowledge of Big Data and related technologies such as Hadoop, Spark, and Airflow

Preferred qualifications, capabilities, and skills

• Background in high performance computing and ML hardware acceleration
• Track record of contributing to open-source ML frameworks
• Knowledge of Kubernetes ecosystem, including EKS, Helm, and Custom Operators

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
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.