Same issue with iMac and Monterey. Keyboard is laggy until I plug it in. Plugging it in solves the problem, no work arounds have fixed the issue yet.