Nubank

Staff Machine Learning Engineer

Nubank
Apply
3 months ago
Palo Alto, CA, USAStaff+

Base Salary

$230k - $345k/yr

Responsibilities

  • Lead and independently execute complex applied AI/ML research initiatives for credit, recommendation systems, GenAI, and real-time inference.
  • Build and optimize scalable deep learning architectures, including Transformers and GNNs, for production deployment.
  • Design models that satisfy MLOps requirements for latency, interpretability, cost-efficiency, and large-scale operation.
  • Coordinate with Data, Infrastructure, and Product stakeholders to solve ambiguous modeling problems and integrate research into production decision systems.
  • Establish technical standards for experimentation, model evaluation, and code quality.
  • Mentor senior engineers and researchers on deep learning, problem formulation, and research methodology.
  • Participate in hiring interviews, lead ML lifecycle improvement task forces, and contribute to research collaborations or internal papers.

Requirements

  • 5–7+ years of professional experience in applied AI/ML, including delivering research-driven systems into production.
  • Deep expertise in deep learning architectures such as Transformers, multimodal models, or GNNs.
  • Strong Python coding skills and proficiency with PyTorch, JAX, or TensorFlow.
  • Strong understanding of MLOps and the constraints of deploying models at scale.
  • Sophisticated ML problem-formulation and problem-solving skills in situations involving uncertainty, messy data, or limited data.
  • Experience with large-scale experimentation and A/B testing.
  • Ability to communicate complex technical concepts to technical peers and cross-functional stakeholders.

Benefits

  • Base salary range of $230k–$345k, plus potential equity, RSUs, and benefits.
  • Medical, dental, vision, life, and AD&D insurance.
  • Extended maternity and paternity leave.
  • Access to Nucleo learning courses and the NuLanguage language-learning program.
  • NuCare mental health and wellness assistance.
  • 401(k), Health Savings Account, and Flexible Spending Account plans.
  • Work-from-home allowance.
  • Relocation assistance when applicable.
  • Palo Alto hybrid work arrangement requiring office attendance at least twice per week, typically two to three days weekly.

Tech Stack

PythonPyTorchTensorFlow

Categories

Nubank

About Nubank

10,000+ employees

Nubank is a digital banking platform for consumers and small businesses in Latin America, offering a no-fee credit card, accounts, payments, lending, and investments through Nu Asset Management. It monetizes via interchange, interest, and product fees within its app-based ecosystem. Founded in 2013 and headquartered in São Paulo, it serves over 100 million customers across Brazil, Mexico, and Colombia and is listed on the NYSE (ticker: NU).

Contact me