partial_trace
==============================================

.. autofunction:: qujax.partial_trace