Senior/Staff Software Engineer (CUDA Expert) - Remote

Overview

Nu is the world’s largest digital banking platform outside of Asia, serving over 105 million customers across Brazil, Mexico, and Colombia. The company has been leading an industry transformation by leveraging data and proprietary technology to develop innovative products and services. Guided by its mission to fight complexity and empower people, Nu caters to customers’ complete financial journey, promoting financial access and advancement with responsible lending and transparency. The company is powered by an efficient and scalable business model that combines low cost to serve with growing returns. Nu’s impact has been recognized in multiple awards, including Time 100 Companies, Fast Company’s Most Innovative Companies, and Forbes World’s Best Banks.

In Short

• Join the AI Infrastructure Squad to build and scale foundational cloud, data, and AI infrastructure.
• Focus on performance, reliability, and scalability in AI systems.
• Deep experience with GPU programming (CUDA, Triton, or OpenCL) is required.
• Strong understanding of large language model architectures.
• Experience profiling and tuning performance for deep learning workloads.
• Proficiency in Python and C++ is necessary.
• Collaborate with cross-functional teams to enhance AI performance.
• Mentor other engineers and share performance learnings.
• Work in a fast-paced environment with cutting-edge AI infrastructure.
• Remote work with quarterly trips to Sao Paulo for team building.

Requirements

• Deep experience with GPU programming and performance optimization.
• Understanding of memory management and GPU-accelerated inference.
• Experience with PyTorch internals or custom kernel development.
• Hands-on knowledge of low-level optimizations in ML pipelines.
• Experience designing infrastructure that scales across multiple GPUs.
• Ability to work across research and engineering teams.

Benefits

• High-impact, cross-functional work with researchers and ML engineers.
• Top tier medical, dental, and vision insurance.
• 20 days of time off and 14 company holidays.
• Remote work opportunities and work-life balance.
• 401K and saving plans available.
• Learning and development programs offered.