Masked diffusion model in Rust, verified by the compile