Main test test_jax_can_use_the_GPU # test_jax_can_use_the_GPU() Test that Jax can use the GPU if it we have one. test_torch_can_use_the_GPU # test_torch_can_use_the_GPU() Test that torch can use the GPU if it we have one. test_defaults # test_defaults(experiment_config: Config) -> None Test to check what the default values are when not specifying anything on the command- line.