JAX STATE WATCHING OPTIONS