More stable fix for JAX Multinomial#1339

ricardoV94 merged 1 commit intopymc-devs:mainpymc-devs/pytensor:mainfrom

ricardoV94:more_stable_multinomial_jaxricardoV94/pytensor:more_stable_multinomial_jaxCopy head branch name to clipboard