flax icon indicating copy to clipboard operation
flax copied to clipboard
verified publisher icon google

Add intermediate value captures based off hijax.Box

Open IvyZX opened this issue 5 months ago • 0 comments

Master tracker: #4924

The whole feature is still WIP. Feedbacks welcome.

The hijax.Box does not yet work with vmap and scan. And not in some weird cases like double-jit.

IvyZX avatar Sep 04 '25 02:09 IvyZX