jax.profiler.save\_device\_memory\_profile
==========================================

.. currentmodule:: jax.profiler

.. autofunction:: save_device_memory_profile