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

pub mod ops;
pub mod primitives;

pub use primitives::Primitive;

mod shape;
pub use shape::{Dim, Dims, Shape};

mod tensor;
pub use tensor::Tensor;

mod dtype;
pub use dtype::{DType, DynDType, ElemType, F16, F32, F64, U32, U8};

pub mod backend;
pub use backend::Backend;

mod value;
pub use value::{Aux, BasicValue, GenericValue, ModuleValue, Value, ValueSpec};

mod transforms;
pub use transforms::{eval, grad, jvp, raiexpr, value_and_grad, vjp, Func, TensorIter};

pub mod dispatch;

pub mod utils;

mod error;
pub use error::{Error, Result};

pub mod nn;

#[macro_use]
mod macros;