jax.Device
==========

.. currentmodule:: jax

.. autoclass:: Device

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~Device.__init__
      ~Device.addressable_memories
      ~Device.default_memory
      ~Device.get_stream_for_external_ready_events
      ~Device.live_buffers
      ~Device.memory
      ~Device.memory_stats
      ~Device.transfer_from_outfeed
      ~Device.transfer_to_infeed
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~Device.client
      ~Device.device_kind
      ~Device.host_id
      ~Device.id
      ~Device.local_hardware_id
      ~Device.platform
      ~Device.process_index
      ~Device.task_id