← All jobs

Staff Machine Learning Engineer

Nubank

Posted

Company
Nubank
Location
Palo Alto, California
Workplace
hybrid
Category
Software Development
Salary
USD 230k–345k/yearly
ab testing gnn jax mlops multimodal python pytorch tensorflow transformers

5-7+ years applied AI/ML delivering research-driven systems into production; deep learning (Transformers/Multimodal/GNNs); Python; PyTorch/JAX/TensorFlow; MLOps; ML problem formulation; A/B testing; communicate complex concepts.

Apply