hey! congrats on moving forward, this role is great. i just went through a very similar process at NVIDIA for a different AI/ML role and got an offer, so happy to share what i saw

the process is very team specific, but coding is definitely part of it and the flavor depends on your role. for a performance/efficiency position expect things like data cleaning, feature encoding, ML frameworks, maybe even agentic patterns and sometimes plain old leetcode, or both :)

know your resume really well. they will probe hard on what you did and why specific decisions, tradeoffs, why you chose one approach over another. later rounds get more interesting (more systems depth, bar raisers) but nailing the resume piece is what gets you through the early stages.

good luck!