jaxsnn.event.types.OptState

class jaxsnn.event.types.OptState(opt_state, weights)

Bases: NamedTuple

__init__()

Initialize self. See help(type(self)) for accurate signature.

Methods

Attributes

opt_state	Alias for field number 0
weights	Alias for field number 1

opt_state: optax.OptState

Alias for field number 0

weights: List[Weight]

Alias for field number 1