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.

Parameters

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

Return type

Optional[str]