JaxBackend.set\_device
==================================

.. currentmodule:: tme.backends

.. automethod:: JaxBackend.set_device