transolver

Classes

Transolver

Implementation of the Transolver model.

Module Contents

class transolver.Transolver(config)

Bases: 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.

Parameters:

config (emmi.schemas.models.TransolverConfig) – Configuration of the Transolver model.