JaxBackend.device_count#

JaxBackend.device_count()#

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

Returns:
int

Number of available devices.