fsdp_config.json

{
    "backward_prefetch": "backward_pre",
    "forward_prefetch": "true",
    "activation_checkpointing": true
}