Nubank

Staff Machine Learning Engineer

Palo Alto, Remote. Posted 3 months ago

The role

You will lead high-impact research projects bridging state-of-the-art AI and production-grade financial systems. You will build and optimize architectures like Transformers and GNNs for use cases such as Credit, RecSys, and GenAI.

What you will do

  • Lead and execute complex applied research initiatives independently
  • Build and optimize architectures like Transformers and GNNs
  • Address difficult and ambiguous modeling problems
  • Design architectures that respect MLOps constraints
  • Develop and deliver innovative solutions for downstream production models
  • Participate in cross-functional collaborations
  • Establish technical standards for experimentation, model evaluation, and code quality
  • Serve as a technical mentor for senior engineers and researchers
  • Participate in hiring interview panels
  • Lead internal task forces to improve the ML lifecycle

Required qualifications

  • 5-7+ years in applied AI/ML
  • Proven track record of delivering research-driven systems into production environments
  • Deep expertise in Deep Learning architectures (Transformers, Multimodal, or GNNs)
  • Strong coding skills in Python
  • Proficiency with frameworks like PyTorch, JAX, or TensorFlow
  • Solid understanding of MLOps and the constraints of deploying models at scale
  • Sophisticated skills in ML problem formulation
  • Ability to navigate uncertainty when data is messy or unavailable
  • Ability to communicate complex technical concepts to technical peers and cross-functional stakeholders
  • Experience with large-scale experimentation and A/B testing

Benefits

  • Opportunity of earning equity at Nu
  • Medical Insurance
  • Dental and Vision Insurance
  • Life Insurance and AD&D
  • Extended maternity and paternity leaves
  • Nucleo - Our learning platform of courses
  • NuLanguage - Our language learning program
  • NuCare - Our mental health and wellness assistance program
  • 401K
  • Saving Plans - Health Saving Account and Flexible Spending Account
  • Work-from-home Allowance
  • Relocation Assistance Package, if applicable

Before you apply

  • Recruitment process may involve the use of artificial intelligence–enabled tools
  • Nu does not provide individualized technical feedback

About the company

Nu serves more than 140 million customers with a mission to fight complexity and empower people. The company uses proprietary technology and data at scale to power its digital platform.

Staff Machine Learning Engineer at Nubank | JobZoey