/atamai

Lightweight ML library in Rust

Primary LanguageRustMIT LicenseMIT

Atamai

Lightweight ML library in Rust.

Activations

  • Linear (alias: Identity);
  • ReLU;
  • Sigmoid (alias: Logistic);
  • Softmax;
  • Tanh.

Losses

  • Binary Cross-Entropy (BCE) (alias: Log Loss);
  • Categorical Cross-Entropy (CCE) (alias: Softmax Loss);
  • Hinge Loss;
  • Mean Absolute Error (MAE) (alias: L1);
  • Mean Squared Error (MSE) (alias: L2);
  • Huber Loss (Smooth Mean Absolute Error (SMAE)).