I'd imagine there's a positive correlation between "uses AI" and productivity, but I expect that to be somewhat moderated by how "well" they use AI.

It's very easy to churn out code that isn't particularly additive (or in some cases, actively destructive), so adjusting for that would be important.