pub fn sample_multinomial(probs: &[f32], rng_state: &mut u64) -> u32
Sample a categorical distribution using a small deterministic LCG. Takes probs (assumed to sum to ~1) and a mutable RNG state, returns a sampled index.
probs