warp icon indicating copy to clipboard operation
warp copied to clipboard

[BUG] Using jax_callable() with GraphMode.WARP results in too many graph captures

Open nvlukasz opened this issue 2 months ago • 0 comments

Bug Description

When JAX doesn't reuse memory for the callable inputs and outputs, a new graph must be captured, which kills perf.

System Information

No response

nvlukasz avatar Oct 21 '25 15:10 nvlukasz