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
#![feature(maybe_uninit_array_assume_init)]
#![feature(fn_traits)]
#![feature(unboxed_closures)]

pub mod ops;
pub mod primitives;
pub use primitives::Primitive;

mod shape;
pub use shape::Shape;

mod tensor;
pub use tensor::Tensor;

mod dtype;
pub use dtype::{DType, ElemType};

pub mod backend;
pub use backend::Backend;

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

pub mod dispatch;

pub mod utils;

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