API
===

horovod.tensorflow
------------------
.. automodule:: horovod.tensorflow

horovod.tensorflow.keras
------------------------
.. automodule:: horovod.tensorflow.keras
.. automodule:: horovod.tensorflow.keras.callbacks
    :special-members: __init__

horovod.keras
-------------
.. automodule:: horovod.keras
.. automodule:: horovod.keras.callbacks
    :special-members: __init__

horovod.torch
-------------
.. automodule:: horovod.torch

horovod.mxnet
-------------
.. automodule:: horovod.mxnet

horovod.spark
-------------
.. automodule:: horovod.spark

horovod.spark.keras
-------------------
.. autoclass:: horovod.spark.keras.KerasEstimator
    :show-inheritance:
.. autoclass:: horovod.spark.keras.KerasModel
    :show-inheritance:

horovod.spark.torch
-------------------
.. autoclass:: horovod.spark.torch.TorchEstimator
    :show-inheritance:
.. autoclass:: horovod.spark.torch.TorchModel
    :show-inheritance:

horovod.spark.common
--------------------
.. autoclass:: horovod.spark.common.estimator.HorovodEstimator
.. autoclass:: horovod.spark.common.estimator.HorovodModel
.. automodule:: horovod.spark.common.backend
    :show-inheritance:
.. automodule:: horovod.spark.common.store
    :show-inheritance:

horovod.run
-------------
.. automodule:: horovod.run