fastnn 0.3.1

A small deep learning library in Rust with optional CUDA kernels
Documentation 
1
2
3
4
5
6
7
8
9
10
11

//! Checkpoints: `.fdl` weights, `.fdl` training state, safetensors, and the
//! 16-bit float conversions safetensors needs.

pub mod checkpoint;
pub mod half;
pub mod safetensors;
pub mod training;

pub use checkpoint::{load, load_tensors, save, save_tensors, Checkpoint};
pub use safetensors::{load_safetensors, load_safetensors_renamed, save_safetensors, LoadReport};
pub use training::{load_training, save_training};