JaxDecomposition
================

.. currentmodule:: fridom.framework.domain_decomposition

.. autoclass:: JaxDecomposition

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~JaxDecomposition.__init__
      ~JaxDecomposition.create_array
      ~JaxDecomposition.create_meshgrid
      ~JaxDecomposition.create_random_array
      ~JaxDecomposition.gather
      ~JaxDecomposition.max
      ~JaxDecomposition.min
      ~JaxDecomposition.pad_extend
      ~JaxDecomposition.pad_trim
      ~JaxDecomposition.parallel_backward_transform
      ~JaxDecomposition.parallel_forward_transform
      ~JaxDecomposition.shard_map
      ~JaxDecomposition.sum
      ~JaxDecomposition.sync
      ~JaxDecomposition.sync_multiple
      ~JaxDecomposition.unpad_extend
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~JaxDecomposition.device_ids
      ~JaxDecomposition.halo
      ~JaxDecomposition.i_am_active
      ~JaxDecomposition.n_dims
      ~JaxDecomposition.p_dims
      ~JaxDecomposition.pad
      ~JaxDecomposition.parallel
      ~JaxDecomposition.periods
      ~JaxDecomposition.rank
      ~JaxDecomposition.shape
      ~JaxDecomposition.shared_axes
      ~JaxDecomposition.size
      ~JaxDecomposition.unpad
   
   

   .. minigallery:: fridom.framework.domain_decomposition.JaxDecomposition
      :add-heading: