JaxBackend.device_count#

JaxBackend.device_count()#

Returns the number of available GPU devices.

On this page
Show Source