ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

Module: oryx.core.state.module

Contains highest-level abstractions for the stateful function API.


class Module: Encapsulates a parameterized function, along with updates to its state.


assign(...): Assigns a value to a variable.

variable(...): Tags a value as a variable.

ASSIGN 'assign'
VARIABLE 'variable'