boostr 0.1.0

ML framework built on numr - attention, quantization, model architectures
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12

pub mod contrastive;
pub mod cross_entropy;
pub mod focal;
pub(crate) mod helpers;
pub mod kl_div;
pub mod mse;

pub use contrastive::contrastive_loss;
pub use cross_entropy::{cross_entropy_loss, cross_entropy_loss_smooth};
pub use focal::focal_loss;
pub use kl_div::kl_div_loss;
pub use mse::mse_loss;