Senior/Staff Software Engineer (Infrastructure)

Nu is seeking a Senior/Staff Software Engineer to join its AI Infrastructure Squad within the AI Core BU. This role involves building and scaling the foundational cloud, data, and AI infrastructure that powers machine learning workloads across the organization. The engineer will design and optimize high-performance training, inference, and data processing systems, ensuring reliability, scalability, and efficiency. Nu values collaborators and mentors who help teammates grow while upholding high engineering standards.

The ideal candidate will thrive in horizontal, high-impact teams and enjoy solving deep technical challenges at the intersection of AI, cloud, and distributed systems. They should also take ownership with a strong product mindset, ensuring infrastructure is reliable, scalable, and built around user needs.

Responsibilities include:

• Leading complex infrastructure projects from design to production.
• Designing high-availability, fault-tolerant systems for AI/ML workloads.
• Working on developer tooling, platform engineering, or ML infrastructure.
• Ensuring AI teams can build and deploy efficiently.
• Hands-on experience with monitoring, observability, and alerting for production systems.

Requirements:

• Strong expertise in cloud infrastructure (AWS or GCP) and distributed computing.
• Experience with Kubernetes, container orchestration, and infrastructure as code (Terraform, Pulumi).
• Proficiency in programming languages. Experience with Python and Go is a plus.
• Experience writing ETL pipelines (experience with Spark or BigQuery is preferred)
• Experience with ML infrastructure, including model training, batch and online inference, and monitoring.
• Strong knowledge of networking, storage, and security in large-scale systems.
• Familiarity with workflow orchestration tools (e.g., Dagster, Airflow) and model-serving frameworks (e.g., Ray Serve, vLLM).
• Experience optimizing performance and cost efficiency of AI workloads on cloud and on-prem environments.

What Nu offers:

• High-Impact, Cross-Functional Work
• Cutting-Edge AI & Cloud Infrastructure
• Growth & Ownership Opportunities
• Culture of Excellence & Collaboration