jax.device\_get
===============

.. currentmodule:: jax

.. autofunction:: device_get