Skip to content

Handle capture variables in nnx.jit. #5490

Open
samanklesaria wants to merge 1 commit into
google:mainfrom
samanklesaria:handle-captures-jit
Open

Handle capture variables in nnx.jit. #5490
samanklesaria wants to merge 1 commit into
google:mainfrom
samanklesaria:handle-captures-jit

Handle captures in jax transforms

d8aa50d
Select commit
Loading
Failed to load commit list.