Expand description
kuji: stochastic sampling primitives.
This crate is meant to be a low-level “sampling toolbox” that other crates can depend on without pulling in domain-specific machinery.
Exposed modules:
reservoir: reservoir sampling (Algorithm L/R) + weighted reservoir.gumbel: Gumbel-max / Gumbel-top-k / relaxed k-hot.neighbor: simple neighborhood sampling helpers (useful for graph ML).
Re-exports§
pub use gumbel::gumbel_max_sample;pub use gumbel::gumbel_noise;pub use gumbel::gumbel_softmax;pub use gumbel::gumbel_topk_sample;pub use gumbel::gumbel_topk_sample_with_rng;pub use gumbel::relaxed_topk_gumbel;pub use neighbor::NeighborSampler;pub use reservoir::ReservoirSampler;pub use reservoir::ReservoirSamplerR;pub use reservoir::WeightedReservoirSampler;