I wrote a little script to do this automatically when your bluetooth keyboard switches between computers. I just updated it to work with m1 Macs based on the recent work.

See https://github.com/mikecarlton/bluekvm