Simple Multi Attention Head Model
==================================

.. automodule:: flood_forecast.transformer_xl.multi_head_base
    :members: