Add a name parameter to the watchdog context manager. #718

Job	Run time
Unittests (Python 3.12, JAX 0.8.2)	27s
Unittests (Python 3.11, JAX 0.8.2)	43s
Unittests (Python 3.11, JAX 0.7.2)	30s
Unittests (Python 3.13, JAX 0.8.0)	28s
Unittests (Python 3.11, JAX 0.9.0)	27s
Unittests (Python 3.13, JAX 0.7.2)	31s
Unittests (Python 3.11, JAX 0.8.0)	26s
Unittests (Python 3.14, JAX 0.8.2)	32s
Unittests (Python 3.14, JAX 0.8.0)	35s
Unittests (Python 3.12, JAX 0.7.2)	30s
Unittests (Python 3.14, JAX 0.7.2)	32s
Unittests (Python 3.12, JAX 0.8.1)	41s
Unittests (Python 3.14, JAX 0.9.0)	34s
Unittests (Python 3.13, JAX nightly)	1m 43s
Unittests (Python 3.13, JAX 0.9.0)	29s
Unittests (Python 3.14, JAX nightly)	1m 48s
Unittests (Python 3.13, JAX 0.8.1)	29s
Unittests (Python 3.12, JAX nightly)	1m 44s
Unittests (Python 3.12, JAX 0.8.0)	31s
Unittests (Python 3.13, JAX 0.8.2)	25s
Unittests (Python 3.12, JAX 0.9.0)	31s
Unittests (Python 3.11, JAX nightly)	1m 38s
Unittests (Python 3.11, JAX 0.8.1)	26s
Unittests (Python 3.14, JAX 0.8.1)	30s
	17m 10s

Provide feedback