Change `decode` to `use_kv_cache` in `genie.py`

[emergenz]

Currently, if decode is set to true, we will use kv caching in the fully causal baseline. This naming was inherited from the nnx API, but is confusing and should be abstracted away.