A RetroSearch Logo

Home - News ( United States | United Kingdom | Italy | Germany ) - Football scores

Search Query:

Showing content from https://docs.jax.dev/en/latest/_autosummary/jax.Device.html below:

jax.Device — JAX documentation

jax.Device#
class jax.Device#

A descriptor of an available device.

Subclasses are used to represent specific types of devices, e.g. CPUs, GPUs. Subclasses may have additional properties specific to that device type.

__init__(*args, **kwargs)#

Methods

Attributes

addressable_memories

Returns all the memories that a device can address.

client

(self) -> object

default_memory

Returns the default memory of a device.

device_kind

(self) -> str

get_stream_for_external_ready_events

host_id

Deprecated; please use process_index

id

Integer ID of this device.

live_buffers

local_hardware_id

Opaque hardware ID, e.g., the CUDA device number.

memory

memory_stats

Returns memory statistics for this device keyed by name.

platform

(self) -> str

process_index

Integer index of this device's process.

task_id

Deprecated; please use process_index

transfer_from_outfeed

transfer_to_infeed


RetroSearch is an open source project built by @garambo | Open a GitHub Issue

Search and Browse the WWW like it's 1997 | Search results from DuckDuckGo

HTML: 3.2 | Encoding: UTF-8 | Version: 0.7.4