with_grad_tape: Run code with gradient tape enabled

In ggmlR: 'GGML' Tensor Operations for Machine Learning

Run code with gradient tape enabled

Description

Records all ag_* operations inside expr for later backward(). When the default device is "gpu", the ggml context is reset at the start of each tape.

Usage

with_grad_tape(expr)

Arguments

expr	Expression to evaluate under gradient tape

Value

Value of last expression in expr (invisibly)

Examples

w <- ag_param(matrix(c(1, 0, 0, 1), 2, 2))
x <- ag_tensor(matrix(c(1, 2), 2, 1))
y <- ag_tensor(matrix(c(1, 2), 2, 1))
with_grad_tape({
  out  <- ag_matmul(w, x)
  loss <- ag_mse_loss(out, y)
})
backward(loss)

ggmlR documentation built on July 14, 2026, 1:08 a.m.