flax
fb391c11 - Prevent child ctxtag from overwriting parent

Commit

66 days ago

Prevent child ctxtag from overwriting parent

References

#5045 - Fix ValueError when `nnx.jit` is used with `nnx.custom_vjp`

Author

samanklesaria

samanklesaria

Committer

samanklesaria

samanklesaria

Parents

Loading