:modulename:`jax_md_mod.model.layers.OutputBlock`
=================================================

.. currentmodule:: jax_md_mod.model.layers

.. autoclass:: OutputBlock

   
   
   .. rubric:: Methods

   .. automethod:: __init__
   .. automethod:: __call__

   .. autosummary::
      :toctree: _autosummary
      :template: method.rst

   
   
   
      ~OutputBlock.params_dict
   
   
      ~OutputBlock.state_dict