Expand description
Axonml Autograd - Automatic Differentiation Engine
§File
crates/axonml-autograd/src/lib.rs
§Author
Andrew Jewell Sr - AutomataNexus
§Updated
March 8, 2026
§Disclaimer
Use at own risk. This software is provided “as is”, without warranty of any kind, express or implied. The author and AutomataNexus shall not be held liable for any damages arising from the use of this software.
Re-exports§
pub use amp::AutocastGuard;pub use amp::AutocastPolicy;pub use amp::autocast;pub use amp::autocast_dtype;pub use amp::disable_autocast;pub use amp::is_autocast_enabled;pub use backward::backward;pub use checkpoint::checkpoint;pub use checkpoint::checkpoint_rng_seed;pub use checkpoint::checkpoint_sequential;pub use functions::FusedAttentionBackward;pub use functions::GruGatesBackward;pub use functions::IdentityBackward;pub use functions::LstmGatesBackward;pub use grad_fn::GradFn;pub use grad_fn::GradientFunction;pub use graph::ComputationGraph;pub use graph::GraphNode;pub use inspect::GraphSnapshot;pub use inspect::SnapshotNode;pub use inspect::depth;pub use inspect::node_count;pub use inspect::to_dot;pub use inspect::trace_backward;pub use no_grad::NoGradGuard;pub use no_grad::no_grad;pub use variable::Variable;
Modules§
- amp
- Automatic Mixed Precision (AMP) Support
- backward
- Backward Pass - Gradient Computation
- checkpoint
- Gradient Checkpointing - Memory-Efficient Training
- functions
- Differentiable Functions - Gradient Implementations
- grad_fn
- Gradient Function Traits - Differentiable Operation Interface
- graph
- Computational Graph - Dynamic Graph Construction
- inspect
- Graph Inspection and Visualization
- no_grad
- No-Grad Context - Disable Gradient Computation
- prelude
- Convenient imports for common autograd usage.
- variable
- Variable - Tensor with Gradient Tracking