Files
rustytorch/demos/rtx-piddm-demo/src/lib.rs
T
2026-03-04 00:08:42 +00:00

17 lines
433 B
Rust

//! PIDDM Demo - Physics-Informed Denoising Diffusion Models.
//!
//! This crate provides the demo backend for the PIDDM demonstration,
//! showing how diffusion models can be trained with physics constraints
//! to generate PDE solutions.
#![forbid(unsafe_code)]
#![warn(missing_docs)]
pub mod data;
pub mod sampling;
pub mod training;
pub use rtx_piddm_shared::*;
pub use sampling::PiddmSampler;
pub use training::PiddmTrainer;