def METHOD_NAME(self):
    a = jt.array(2.0)
    assert a.requires_grad == True
    a.requires_grad = False
    assert a.requires_grad == False
    assert jt.grad(a**2, a) == 0
    a.requires_grad = True
    assert a.requires_grad == True
    assert jt.grad(a**2, a) == 4