Flax - Test

Rewritten flax.jax_utils.prefetch_to_device and flax.jax_utils.replicate using jax.device_put #3391

Sign in to view logs

Run time

Learn about OS pricing on GitHub Actions

Job	Run time
Test import standalone (3.11)	26s
Test import standalone (3.13)	21s
Test pre-commit hooks	20s
Check commit count	8s
Test import standalone (3.12)	27s
Run Tests (3.11, pytest, newest)	2m 52s
Run Tests (3.12, doctest, newest)	4m 27s
Run Tests (3.11, doctest, newest)	4m 11s
Run Tests (3.12, mypy, newest)	1m 13s
Run Tests (3.12, pytest, newest)	2m 49s
Run Tests (3.12, pytype, newest)	11m 0s
Run Tests (3.13, doctest, newest)	4m 7s
Run Tests (3.13, pytest, newest)	2m 54s
	35m 15s