Merge pull request #210 from AI-Hypercomputer:wstcliyu/release-v017 #778

Job	Run time
Unittests (Python 3.11, JAX 0.8.1)	24s
Unittests (Python 3.13, JAX 0.8.3)	32s
Unittests (Python 3.11, JAX 0.8.3)	28s
Unittests (Python 3.14, JAX 0.8.2)	32s
Unittests (Python 3.14, JAX nightly)	2m 10s
Unittests (Python 3.12, JAX 0.9.2)	29s
Unittests (Python 3.14, JAX 0.8.3)	29s
Unittests (Python 3.12, JAX 0.8.0)	31s
Unittests (Python 3.11, JAX 0.9.1)	30s
Unittests (Python 3.12, JAX 0.8.3)	31s
Unittests (Python 3.13, JAX 0.8.0)	28s
Unittests (Python 3.11, JAX 0.8.0)	27s
Unittests (Python 3.11, JAX nightly)	1m 33s
Unittests (Python 3.12, JAX 0.8.1)	31s
Unittests (Python 3.13, JAX 0.8.1)	32s
Unittests (Python 3.14, JAX 0.9.2)	36s
Unittests (Python 3.13, JAX 0.8.2)	33s
Unittests (Python 3.11, JAX 0.9.0)	25s
Unittests (Python 3.14, JAX 0.9.0)	33s
Unittests (Python 3.12, JAX 0.8.2)	33s
Unittests (Python 3.12, JAX 0.9.0)	33s
Unittests (Python 3.14, JAX 0.8.0)	32s
Unittests (Python 3.14, JAX 0.9.1)	32s
Unittests (Python 3.11, JAX 0.8.2)	25s
Unittests (Python 3.14, JAX 0.8.1)	35s
Unittests (Python 3.12, JAX 0.9.1)	27s
Unittests (Python 3.12, JAX nightly)	1m 35s
Unittests (Python 3.13, JAX 0.9.2)	31s
Unittests (Python 3.13, JAX 0.9.1)	25s
Unittests (Python 3.11, JAX 0.9.2)	26s
Unittests (Python 3.13, JAX 0.9.0)	30s
Unittests (Python 3.13, JAX nightly)	1m 35s
	20m 53s

Provide feedback