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

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

jax.device_count — JAX documentation

jax.device_count(backend=None)[source]#

Returns the total number of devices.

On most platforms, this is the same as jax.local_device_count(). However, on multi-process platforms where different devices are associated with different processes, this will return the total number of devices across all processes.

Parameters:

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'.

Returns:

Number of devices.

Return type:

int

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