fix: prevent zero-probability events from being sampled in multinomial function in Jax-backend

fix: prevent zero-probability events from being sampled in multinomial function in Jax-backend #464