reax.Engine
===========

.. currentmodule:: reax

.. autoclass:: Engine

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~Engine.__init__
      ~Engine.all_reduce
      ~Engine.barrier
      ~Engine.broadcast
      ~Engine.call
      ~Engine.compute
      ~Engine.default_device
      ~Engine.finalize
      ~Engine.profile
      ~Engine.setup
      ~Engine.setup_dataloaders
      ~Engine.to_device
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~Engine.default_root_dir
      ~Engine.device
      ~Engine.is_global_zero
      ~Engine.local_process_index
      ~Engine.logger
      ~Engine.loggers
      ~Engine.node_rank
      ~Engine.process_count
      ~Engine.process_index
      ~Engine.profiler
      ~Engine.rngs
      ~Engine.strategy