===================
Halo Exchange Layer
===================

Overview
========

.. automodule:: distdl.nn.halo_exchange

Halo Mixin
----------

.. automodule:: distdl.nn.mixins.halo_mixin

API
===

.. currentmodule:: distdl.nn

.. autoclass:: HaloExchange
    :members:
    :exclude-members: forward


.. currentmodule:: distdl.nn.mixins

.. autoclass:: HaloMixin
    :members: