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

//! Standard SSL data augmentation helpers operating on `[C, H, W]` CHW tensors.

pub mod color;
pub mod multi_crop;
pub mod rand_augment;
pub mod solarize_blur;

pub use rand_augment::{
    AugOp, AutoAugPolicy, AutoAugmentConfig, RandAugmentConfig, SubPolicy, all_aug_ops,
    apply_aug_op, auto_augment, rand_augment,
};
pub use solarize_blur::{
    SimClrBlurSolarConfig, add_gaussian_noise, gaussian_blur_chw, random_gaussian_blur_chw,
    random_solarize, simclr_blur_solar, solarize,
};