Gain Components

class tabascal.components.gains.BaseGPGains

build_set_params()

Build parameter sampling function (optional)

setup(tab_config: TabConfig)

Initialize component with configuration

class tabascal.components.gains.GPGains

build_constants()

Return arrays that do not change during the forward pass.

Returns a dict of array_name -> array_value. These will be stored in constants as “_c/<ClassName>/array_name” by Model.__init__.

build_forward()

Return pure, JIT-compatible function

build_set_params()

Build parameter sampling function (optional)

setup(tab_config: TabConfig)

All validation and error-prone operations here

class tabascal.components.gains.UnitaryGains

build_forward()

Build the forward computation function

setup(tab_config: TabConfig)

All validation and error-prone operations here