NodeComponent
NodeComponent#
- class liesel.liesel.nodes.NodeComponent(*args, **kwargs)[source]#
Bases:
abc.ABC
An abstract node component with inputs and a NumPy module.
Superclass of the
NodeDistribution
and theNodeCalculator
.Methods
jaxify
()Enables JAX NumPy for the node component.
unjaxify
()Disables JAX NumPy for the node component.
Attributes
The inputs of the node component.
Whether JAX NumPy is enabled for the node component.