oxicuda-ssl 0.2.0

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
10
11
12

//! 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 dino_v2;
pub mod ibot;
pub mod swav;

pub use dino_v2::{DinoV2, DinoV2Config};