.. role:: hidden
    :class: hidden-section

dist_model_parallel
===================================
.. currentmodule:: distributed_embeddings.python.layers.dist_model_parallel


:hidden:`DistributedEmbedding`
~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~
.. autoclass:: DistributedEmbedding
    :members: get_weights, set_weights

.. autofunction:: broadcast_variables

.. autofunction:: DistributedGradientTape