jax.profiler.device\_memory\_profile
====================================

.. currentmodule:: jax.profiler

.. autofunction:: device_memory_profile