kuji 0.1.9

Stochastic sampling primitives: Gumbel-Softmax, reservoir sampling, and latent permutations
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31

//! `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).
//! - `qmc`: quasi-Monte Carlo sequences (Sobol, Halton, Owen-scrambled Sobol).
//! - `thinning`: kernel thinning and herding (greedy coreset selection via MMD).

#![forbid(unsafe_code)]
#![warn(missing_docs)]

pub mod gumbel;
pub mod neighbor;
pub mod qmc;
pub mod reservoir;
pub mod tconorm;
pub mod thinning;

pub use gumbel::{
    gumbel_max_sample, gumbel_noise, gumbel_softmax, gumbel_topk_sample,
    gumbel_topk_sample_with_rng, relaxed_topk_gumbel,
};
pub use neighbor::NeighborSampler;
pub use qmc::{halton_point, halton_sequence, sobol_scrambled, sobol_sequence, SobolGenerator};
pub use reservoir::{ReservoirSampler, ReservoirSamplerR, WeightedReservoirSampler};
pub use tconorm::{tconorm, tconorm_fold, tnorm, tnorm_fold, TConormFamily};
pub use thinning::{kernel_herd, kernel_thin, mmd_sq_from_gram};