Remove LEGACY support from checkpointing to simplify path

Signed-off-by: Ananth Subramaniam <ansubramania@nvidia.com>

Author: ananthsub