Lead Software Engineer - AI/ML Platform

Key Skills

Argo, Azure, Continuous Integration & Continuous Delivery - CI/CD, Django, ETL frameworks, Flask, GraphQL, Hubspot, JavaScript, Jenkins, Jupyter Notebook, Kibana, Kubernetes-K8s, Node.js, NoSQL, PostgreSQL, Python Programming, PyTorch, REST API, Spark-SQL, SQL, Swagger, TensorFlow, Workfusion

Job Description

What you will do:

• Build, modernize and maintain the U.S. Bank AI/ML Platform & related frameworks / solutions.
• Participate and contribute in architecture & design reviews.
• Build/Deploy AI/ML platform in Azure with open-source applications (Argo, Jupyter Hub/Kubeflow) and/or cloud/SaaS solutions (Azure ML, Databricks).
• You will design, develop, test, deploy, and maintain distributed & GPU-enabled Machine Learning Pipelines using K8s/AKS based Argo Workflow Orchestration solutions, while collaborating with Data Scientists.
• Enable/Support platform to do distributed data processing using Apache Spark and other distributed / scale technologies.
• Build ETL pipelines, ingress / egress methodologies in context to AIML use-cases.
• Build highly scalable backend REST APIs for metadata management and other misc. business needs.
• Deploy Application in Azure Kubernetes Service using GitLab, Jenkins, Docker, Kubectl, Helm and Manifest
• Experience in branching, tagging, and maintaining the versions across different environments in GitLab.
• Review code developed by other developers and provide feedback to ensure best practices (e.g., design patterns, accuracy, testability, efficiency etc.)
• Work with relevant engineering, operations, business lines, and infrastructure groups to ensure effective architectures and designs and communicate findings clearly to technical and non-technical partners.
• Perform functional, benchmark & performance testing and tuning to achieve performant AIML workflow(s), interactive notebook user experiences, and pipelines.
• Assess, design & optimize the resources capacities for ML based resource (GPU) intensive workloads.
• Communicate processes and results of the application with all parties involved in the product team, like engineers, product owner, scrum master and third-party vendors.

Basic Qualifications:

• Bachelor’s/master’s degree in computer science or data science
• 6 – 8 years of experience in software development and with data structures/algorithms

Preferred Qualifications / Skills:

• Experience developing AIML platforms & frameworks (including core offerings such as model training, inferencing, distributed/parallel programming), preferably on Kubernetes and native cloud.
• Highly skilled with Python or JAVA programming languages
• Highly skilled with database languages like SQL & NoSQL
• Experience designing, developing, and deploying highly maintainable, extensible, and testable distributed applications using Python and other languages.
• Experience developing ETL pipelines and REST APIs in Python using Flask or Django
• Experienced with technologies/frameworks including Kubernetes, Helm Charts, Notebooks, Workflow orchestration tools, and CI/CD & monitoring frameworks.
• Experience with AI/ML open-source projects in large datasets using Jupyter, Argo, Spark, Pytorch, TensorFlow
• Experience creating Unit and Functional test cases using PyTest, UnitTest
• Experience with training and tuning models in Machine Learning
• Experience working with Jupyter Hub
• Experience with DB management system like PostgreSQL
• Experience in searching, monitoring, and analyzing logs using Splunk/Kibana
• GraphQL/Swagger implementation knowledge
• Strong understanding and experience with Kubernetes for availability and scalability of applications in Azure Kubernetes Service
• Experience building CI/CD pipelines using Cloudbees Jenkins, Docker, Artifactory, Kubernetes, Helm Charts and Gitlab
• Experience with tools like Jupyter Hub, Kubeflow, MLFlow, TensorFlow, Scikit, Apache Spark, Kafka
• Experience with workflow orchestration tools such as Apache Airflow, Argo workflows
• Familiarity with Conda, PyPi, and Node.js package builds

If there’s anything we can do to accommodate a disability during any portion of the application or hiring process, please refer to our disability accommodations for applicants.

Learn how the way we work at U.S. Bank drives meaningful relationships with our customers and collaboration across the company.

Benefits:

Our approach to benefits and total rewards considers our team members’ whole selves and what may be needed to thrive in and outside work. That's why our benefits are designed to help you and your family boost your health, protect your financial security and give you peace of mind. Our benefits include the following (some may vary based on role, location or hours):

• Healthcare (medical, dental, vision)

• Basic term and optional term life insurance

• Short-term and long-term disability

• Pregnancy disability and parental leave

• 401(k) and employer-funded retirement plan

• Paid vacation (from two to five weeks depending on salary grade and tenure)

• Up to 11 paid holiday opportunities

• Adoption assistance

• Sick and Safe Leave accruals of one hour for every 30 worked, up to 80 hours per calendar year unless otherwise provided by law

EEO is the Law

U.S. Bank is an equal opportunity employer committed to creating a diverse workforce. We consider all qualified applicants without regard to race, religion, color, sex, national origin, age, sexual orientation, gender identity, disability or veteran status, among other factors. Applicants can learn more about the company’s status as an equal opportunity employer by viewing the federal KNOW YOUR RIGHTS EEO poster. 

E-Verify

U.S. Bank participates in the U.S. Department of Homeland Security E-Verify program in all facilities located in the United States and certain U.S. territories. The E-Verify program is an Internet-based employment eligibility verification system operated by the U.S. Citizenship and Immigration Services. Learn more about the E-Verify program.

The salary range reflects figures based on the primary location, which is listed first. The actual range for the role may differ based on the location of the role. In addition to salary, US Bank offers a comprehensive benefits package, including incentive and recognition programs, equity stock purchase 401k contribution and pension (all benefits are subject to eligibility requirements). Pay Range: $139,995.00 - $164,700.00 - $181,170.00