US considers India a partner in non-proliferation