ferrotorch-core 0.1.3

Core tensor and autograd engine for ferrotorch — PyTorch in Rust
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

pub mod autocast;
pub mod autocast_ops;
pub mod checkpoint;
pub mod fixed_point;
pub mod grad_penalty;
pub mod graph;
pub mod higher_order;
pub mod no_grad;

pub use autocast::{autocast, autocast_dtype, is_autocast_enabled, AutocastDtype};
pub use autocast_ops::{
    autocast_category, autocast_log, should_cast_to_reduced, should_keep_full_precision,
    AutocastCategory,
};
pub use fixed_point::fixed_point;
pub use grad_penalty::{grad_norm, gradient_penalty, jvp, vjp};
pub use graph::{backward, backward_with_grad};
pub use higher_order::{grad, hessian, jacobian};
pub use no_grad::{enable_grad, is_grad_enabled, no_grad, set_grad_enabled};