Hi, I was wondering if it would be possible to save the compiled GradientTape that is used in ReverseDiff.jl. Using the gradient example from the package, I’d like to save compiled_f_tape # some objective function to work with f(a, b) = sum(a' * b + a * b') # pre-record a GradientTape for `f` …

Possible to save compiled GradientTape after compile in ReverseDiff

xxxxx March 8, 2023, 12:24am 7

The number of parameters is around 50,000, and the size of the data is about 20M.

Topic		Replies	Views
Unexpected performance mismatch in gradients for "compiled-tape-in-tape" experiment Performance autodiff	0	196	June 25, 2023
ReverseDiff.GradientTapes for functions of the form x -> f(x, y) when gradient is to be calculated wrt x only? General Usage	2	832	November 4, 2019
Save module/function using PackageCompiler General Usage package-compiler	7	144	January 16, 2026
ReverseDiff Incorrect Result Numerics	5	607	July 19, 2018
Repeatedly calculating a hessian using both ForwardDiff and ReverseDiff Optimization (Mathematical) question , differentiation	3	1554	February 2, 2021