Crate autograph

source ·
Expand description

A machine learning library for Rust.

GPGPU kernels implemented with krnl.

  • Host and device execution.
  • Tensors emulate ndarray.
    • Host tensors can be borrowed as arrays.
  • Tensors, models, and optimizers can be serialized with serde.
    • Portable between platforms.
    • Save and resume training progress.
  • Fully extensible, in Rust.

Re-exports§

Modules§