jax.experimental.pallas.tpu.CompilerParams
==========================================

.. currentmodule:: jax.experimental.pallas.tpu

.. autoclass:: CompilerParams

   
   .. automethod:: __init__

   
   .. rubric:: Methods

   .. autosummary::
   
      ~CompilerParams.__init__
      ~CompilerParams.replace
   
   

   
   
   .. rubric:: Attributes

   .. autosummary::
   
      ~CompilerParams.allow_collective_id_without_custom_barrier
      ~CompilerParams.allow_input_fusion
      ~CompilerParams.collective_id
      ~CompilerParams.dimension_semantics
      ~CompilerParams.disable_bounds_checks
      ~CompilerParams.disable_semaphore_checks
      ~CompilerParams.flags
      ~CompilerParams.fuse_transposed_lhs_in_matmul
      ~CompilerParams.has_side_effects
      ~CompilerParams.internal_scratch_in_bytes
      ~CompilerParams.needs_layout_passes
      ~CompilerParams.serialization_format
      ~CompilerParams.shape_invariant_numerics
      ~CompilerParams.skip_device_barrier
      ~CompilerParams.use_tc_tiling_on_sc
      ~CompilerParams.vmem_limit_bytes