def METHOD_NAME():
    a = shared(0)
    aesara_jax_fn = function([], a, updates={a: a + 1}, mode="JAX")
    res1, res2 = aesara_jax_fn(), aesara_jax_fn()
    assert res1 == 0
    assert res2 == 1
    assert a.get_value() == 2
    a.set_value(5)
    res1, res2 = aesara_jax_fn(), aesara_jax_fn()
    assert res1 == 5
    assert res2 == 6
    assert a.get_value() == 7