tidu 0.2.0

Automatic-differentiation transforms (linearize, transpose, eager reverse-mode) for primitive computation graphs.
Documentation
1
2
3
4
5
6
7
8
9
10
11
12
//! Generic eager reverse-mode recording for `Primitive` frontends.

mod backward;
mod record;
mod trace;

pub use backward::{backward, BackwardExecutor};
pub use record::{
    EagerInput, EagerOutput, EagerRecordError, EagerRecordResult, KeySource, RecordedGraph,
    Recorder,
};
pub use trace::Trace;