oxicuda-seq 0.3.0

OxiCUDA: Sequence Models & Structured Prediction (HMM/CRF/Kalman/MRF/alignment)
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
//! Differentiable structured-prediction layers.
//!
//! Currently the [`sinkhorn_crf`] module: entropy-regularised optimal-transport
//! (Sinkhorn) normalisation as a differentiable replacement for sum-product
//! normalisation in matching / permutation structured prediction.

pub mod sinkhorn_crf;

pub use sinkhorn_crf::{
    SinkhornConfig, SinkhornCrf, SinkhornResult, sinkhorn_normalize,
    sinkhorn_normalize_with_margins,
};