|View source on GitHub|
Tags a value as a variable.
oryx.core.state.variable( value, *, name: str, key=None, mode: str = 'strict' )
variable should be used to initialize state in stateful functions.
variable will be called with a value downstream of an
initialization key. The
init transformation will then pull all values tagged
as variables in a function body and store them in a
||JAX value to be tagged as variable.|
||string name for the value.|
JAX value that is used to tie in
string name for sow mode (see
|The value that was passed in.|