JaxBackend.device_count

JaxBackend.device_count()

Return the number of available compute devices considered by the backend.

Returns:

int

Number of available devices.