oxicuda-ssl 0.1.8

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
6
7
8
9
//! Clustering SSL losses: SwAV (Sinkhorn-Knopp normalised assignments),
//! DINO (centred + sharpened student-teacher cross-entropy),
//! iBOT (masked image modeling with online tokenizer), and
//! DeepCluster / DeeperCluster (k-means pseudo-label clustering).

pub mod deep_cluster;
pub mod dino;
pub mod ibot;
pub mod swav;