topos 0.13.0

An autodiff compiler stack in Rust.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70

//! The differentiable operation set: the [`Op`] node vocabulary
//! the graph records and the executor replays, one file per
//! operation, each a pure forward/backward rule over the payload
//! traits.

mod add;
mod broadcast;
mod broadcast_along;
mod div;
mod fold;
mod gather;
mod input;
mod leaf;
mod log_softmax;
mod log_sum_exp;
mod map;
mod matmul;
mod maximum;
mod mul;
mod narrow;
mod neg;
// The module convention names each file after its main concept, and this
// module's main concept is the `Op` enum itself; the inception is
// deliberate.
#[allow(clippy::module_inception)]
mod op;
mod operation;
mod pad;
mod parameter;
mod permute;
mod powf;
mod reshape;
mod scatter;
mod slot;
mod step;
mod sub;
mod sum;
mod sum_along;
mod unfold;

pub(crate) use add::Add;
pub(crate) use broadcast::Broadcast;
pub(crate) use broadcast_along::BroadcastAlong;
pub(crate) use div::Div;
pub(crate) use fold::Fold;
pub(crate) use gather::Gather;
pub(crate) use input::Input;
pub(crate) use leaf::Leaf;
pub(crate) use log_softmax::LogSoftmax;
pub(crate) use log_sum_exp::LogSumExp;
pub(crate) use map::Map;
pub(crate) use matmul::MatMul;
pub(crate) use maximum::Maximum;
pub(crate) use mul::Mul;
pub(crate) use narrow::Narrow;
pub(crate) use neg::Neg;
pub(crate) use op::Op;
pub(crate) use operation::{Cotangents, Operation, Reads, binary, unary};
pub(crate) use pad::Pad;
pub(crate) use parameter::Parameter;
pub(crate) use permute::Permute;
pub(crate) use powf::Powf;
pub(crate) use reshape::Reshape;
pub(crate) use scatter::Scatter;
pub(crate) use slot::SlotId;
pub(crate) use step::Step;
pub(crate) use sub::Sub;
pub(crate) use sum::Sum;
pub(crate) use sum_along::SumAlong;
pub(crate) use unfold::Unfold;