Crate autograph

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

pub extern crate anyhow;

pub extern crate half;

pub extern crate krnl;

pub extern crate ndarray;

pub extern crate num_traits;

Modules

datasetdataset

Datasets.

learnlearn

Machine Learning.

ops

Ops.

tensor

Tensors.