Yes, you can, and should, use your New Jersey address and New Jersey state ID on your New York amended return. Any correspondence concerning your amended return (possibly including a mailed refund) will be sent to the mailing address on the amended return. So it has to be your current address.
