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