JPMorgan Senior Lead Software Engineer - Machine Learning 

United States, New Jersey, Jersey City 

635494276

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

• 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