brainstate.transform.optimize\_jaxpr
====================================

.. currentmodule:: brainstate.transform

.. autofunction:: optimize_jaxpr