@mr_penguin were you able to solve this problem? Bambi has a multinomial family now which works quite similar to brms.
brms