Lead ML Data Engineer, AI Core

About the role

Nu is a digital financial platform serving over 127 million customers in Brazil, Mexico, and Colombia. We combine technology, data, and efficiency to offer simple, accessible, and human financial products.

Key Responsibilities

• Design and build scalable data ingestion pipelines for AI Core, ensuring reliable and efficient data flow.
• Implement data quality monitoring and validation systems to catch issues early.
• Model new types of data into foundation models and analyze their impact on existing models.
• Develop and maintain data preparation workflows using distributed computing frameworks like Ray.
• Tune and optimize machine learning models when new datasets are integrated.
• Collaborate with AI Core ML, Platform, and Infra teams to ensure seamless data flow across the ML infrastructure.
• Lead technical initiatives to improve data engineering practices and mentor team members.

Basic Qualifications

• Typically 6+ years of experience in machine learning engineering, data engineering, or related fields.
• Proven experience designing and building data ingestion pipelines at scale.
• Strong background in applied machine learning, including model training, hyperparameter tuning, and performance evaluation.
• Experience analyzing how data changes impact model performance and designing experiments to measure improvements.
• Proficiency in Python for data engineering and ML workflows, with experience working with large-scale data processing systems.
• Solid understanding of data quality principles and experience implementing monitoring, validation, and alerting systems.
• Strong problem-solving skills and the ability to address complex, ambiguous problems.
• Excellent communication skills, capable of explaining technical concepts to both technical and non-technical stakeholders.
• Demonstrated leadership experience, including mentoring team members and contributing to technical decision-making.

Preferred Qualifications

• Experience with MLflow or similar model tracking and versioning systems.
• Knowledge of foundation models, fine-tuning workflows, and transformer architectures.
• Experience with data pipeline orchestration tools (Dagster, Airflow, or similar).
• Background in financial services or fintech, understanding unique data challenges.
• Experience working in a fast-paced, high-growth environment with distributed teams.
• Track record of reducing complexity in data systems and improving developer experience for ML teams.

Benefits

• Opportunity to earn equity at Nu
• Medical Insurance
• Dental and Vision Insurance
• Life Insurance and AD&D
• Extended maternity and paternity leaves
• Nucleo - Learning platform of courses
• NuLanguage - Language learning program
• NuCare - Mental health and wellness assistance program
• Extended maternity and paternity leaves
• 401K
• Saving Plans - Health Saving Account and Flexible Spending Account
• Work-from-home allowance
• Relocation Assistance Package, if applicable.

Work Model

This role is available in any of our North American offices (Palo Alto, USA; Miami, USA; Durham, USA; Toronto, CAN)