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

//! Projection and predictor MLP heads used by SSL pipelines.

pub mod linear_probe;
pub mod predictor;
pub mod projector;

pub use linear_probe::{
    FittedLinearProbe, LinearProbeConfig, LinearProbeResult, linear_probe_eval, linear_probe_fit,
    linear_probe_predict,
};