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.

NodeID

Unique identifier generated for each node.