autodiff

Overview ¶

Package autodiff provides automatic differentiation capabilities.

This package implements reverse-mode automatic differentiation (backpropagation) using a gradient tape. It wraps any backend to add autodiff capabilities.

Example:

import (
    "github.com/born-ml/born/autodiff"
    "github.com/born-ml/born/backend/cpu"
    "github.com/born-ml/born/tensor"
)

func main() {
    // Wrap CPU backend with autodiff
    base := cpu.New()
    backend := autodiff.New(base)

    // Use for training
    x := tensor.Zeros[float32](tensor.Shape{2, 3}, backend)
    y := x.Add(x)  // Operations recorded on tape

    // Compute gradients
    grads := backend.Backward(y.Raw())
}