jax.print_environment_info

jax.print_environment_info#

jax.print_environment_info(return_string=False)[source]#

Returns a string containing local environment & JAX installation information.

This is useful information to include when asking a question or filing a bug.

Args: return_string (bool) : if True, return the string rather than printing to stdout.

Parameters:

return_string (bool) –

Return type:

str | None