oxicuda-ssl 0.1.7

Self-supervised learning primitives for OxiCUDA: SimCLR, MoCo, BYOL, Barlow Twins, VICReg, MAE, SwAV, DINO — pure Rust, zero CUDA SDK dependency.
Documentation
1
2
3
4
5
//! Clustering SSL losses: SwAV (Sinkhorn-Knopp normalised assignments) and
//! DINO (centred + sharpened student-teacher cross-entropy).

pub mod dino;
pub mod swav;