Some eval and profile routines for fsdp
This runs an fsdp CNN model for a default of 2 epochs to provide a review framework for FSDP.
python fsdp_mnist.py <optional flags>
-- use_offload = Activates cpu_offload
-- profile = Run PyTorch profile tracing
-- save-model = Save model (only working on nightlies)
