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
13
14
15

//! High-level SSL model structs that own their own weights.
//!
//! This module collects struct-based counterparts to the functional APIs
//! scattered across the other `oxicuda-ssl` submodules. Each struct here owns
//! its weight matrices and provides an ergonomic interface for construction,
//! forward passes, and loss computation without requiring the caller to manage
//! raw weight buffers.

pub mod data2vec_v2;
pub mod jem;
pub mod sim_siam;

pub use data2vec_v2::{Data2VecModel, Data2VecModelConfig};
pub use jem::{Jem, JemConfig};
pub use sim_siam::{SimSiam, SimSiamConfig};