Crate burn_autodiff

Burn Autodiff

This autodiff library is a part of the Burn project. It is a standalone crate that can be used to perform automatic differentiation on tensors. It is designed to be used with the Burn Tensor crate, but it can be used with any tensor library that implements the Backend trait.

Modules

• checkpoint
  Checkpoint module.
• grads
  Gradients module.
• ops
  Operation module.

Macros

• retro_binary
  Creates a RetroForward struct for binary operations
• retro_unary
  Creates a RetroForward struct for unary scalar operations
• retro_unary_scalar
  Creates a RetroForward struct for unary scalar operations

Structs

• Autodiff
  Enable auto-differentiation on a backend.
• AutodiffBridge
  Enable autodiff on a backend bridge.
• NodeID
  Unique identifier generated for each node.