Home - News ( United States | United Kingdom | Italy | Germany ) - Football scores

Showing content from https://jax.readthedocs.io/en/latest/_autosummary/jax.local_devices.html below:

jax.local_devices — JAX documentation

jax.local_devices(process_index=None, backend=None, host_id=None)[source]#

Like jax.devices(), but only returns devices local to a given process.

If process_index is None, returns devices local to this process.

Parameters:

• process_index (int | None) – the integer index of the process. Process indices can be retrieved via len(jax.process_count()).

• backend (str | xla_client.Client | None) – This is an experimental feature and the API is likely to change. Optional, a string representing the xla backend: 'cpu', 'gpu', or 'tpu'.

• host_id (int | None)

Returns:

List of Device subclasses.

Return type:

list[xla_client.Device]

RetroSearch is an open source project built by @garambo | Open a GitHub Issue

Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo

HTML: 3.2 | Encoding: UTF-8 | Version: 0.7.5