transolver
==========

.. py:module:: transolver


Classes
-------

.. autoapisummary::

   transolver.Transolver


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

.. py:class:: Transolver(config)

   Bases: :py:obj:`emmi.models.Transformer`


   Implementation of the Transolver model.
   Reference code: https://github.com/thuml/Transolver/
   Paper: https://arxiv.org/abs/2402.02366
   Transolver is a Transformer with a special physics attention mechanism. Hence we extend the Transformer class, and configure it accordingly.


   :param config: Configuration of the Transolver model.