Wrap torch checkpoint() fn to default use_reentrant flag to False and allow env var override

Wrap torch checkpoint() fn to default use_reentrant flag to False and allow env var override #2432