jax.tree\_util.flatten\_one\_level\_with\_keys
==============================================

.. currentmodule:: jax.tree_util

.. autofunction:: flatten_one_level_with_keys