upt_config
==========

.. py:module:: upt_config


Classes
-------

.. autoapisummary::

   upt_config.UPTConfig


Module Contents
---------------

.. py:class:: UPTConfig(/, **data)

   Bases: :py:obj:`pydantic.BaseModel`


   Configuration for a Transolver model.

   Create a new model by parsing and validating input data from keyword arguments.

   Raises [`ValidationError`][pydantic_core.ValidationError] if the input data cannot be
   validated to form a valid model.

   `self` is explicitly positional-only to allow `self` as a field name.


   .. py:attribute:: num_heads
      :type:  int
      :value: None



   .. py:attribute:: hidden_dim
      :type:  int
      :value: None



   .. py:attribute:: mlp_expansion_factor
      :type:  int
      :value: None



   .. py:attribute:: encoder_depth
      :type:  int
      :value: None



   .. py:attribute:: use_rope
      :type:  bool
      :value: None



   .. py:attribute:: supernode_pooling_config
      :type:  emmi.schemas.modules.encoder.SupernodePoolingConfig


   .. py:attribute:: approximator_config
      :type:  emmi.schemas.modules.blocks.TransformerBlockConfig


   .. py:attribute:: decoder_config
      :type:  emmi.schemas.modules.decoder.DeepPerceiverDecoderConfig


   .. py:attribute:: bias_layers
      :type:  bool
      :value: None