jaxlib.xla\_extension.GpuDevice
===============================

.. currentmodule:: jaxlib.xla_extension

.. autoclass:: GpuDevice

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~GpuDevice.__init__
      ~GpuDevice.live_buffers
      ~GpuDevice.transfer_from_outfeed
      ~GpuDevice.transfer_to_infeed
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~GpuDevice.client
      ~GpuDevice.device_kind
      ~GpuDevice.device_vendor
      ~GpuDevice.host_id
      ~GpuDevice.id
      ~GpuDevice.platform
      ~GpuDevice.process_index
      ~GpuDevice.task_id