flax
4f5d6fb9 - Add conditionals `nnx.while_loop` and `nnx.switch`

Commit

1 year ago

Add conditionals `nnx.while_loop` and `nnx.switch`

References

#4343 - Add NNX transforms `nnx.while_loop` and `nnx.switch`

Author

IvyZX

IvyZX

Parents

Loading