Skip to content

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.