Mazda dealer refused to change transmission fluid

OP, earlier in this thread you mentioned the warranty would run out February 2023. Were you able to get a satisfactory resolution ?
If it was me given this was in warranty, I would have just complained about the shifts. At least get it documented. Then let them do whatever steps to fix it.