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.