Add NNX transforms `nnx.while_loop` and `nnx.switch` #530

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Test import standalone (3.10)	12s
Test import standalone (3.11)	11s
Cancel Previous Runs	3s
Check commit count	8s
Test pre-commit hooks	23s
Run Tests (3.10, doctest, newest)	4m 21s
Run Tests (3.10, pytest, newest)	4m 7s
Run Tests (3.10, mypy, newest)	1m 43s
Run Tests (3.11, doctest, newest)	4m 17s
Run Tests (3.11, pytest, newest)	4m 8s
Run Tests (3.11, pytype, newest)	8m 39s
Run Tests (3.10, pytest, 0.4.27)	4m 9s
	32m 21s