code: def MLP_FSDP(
      model_config: AttrDict,
      dims: List[int],
      use_bn: bool = False,
      use_relu: bool = False,
      use_dropout: bool = False,
      use_bias: bool = True,
      skip_last_layer_relu_bn: bool = True,
  ):
      mlp = MLP(
          model_config,
          dims,
          use_bn,
          use_relu,
          use_dropout,
          use_bias,
          skip_last_layer_relu_bn,
      )
      mlp = fsdp_auto_wrap_bn(mlp)
      return fsdp_wrapper(mlp, **model_config.FSDP_CONFIG)
  
file_path: vissl/vissl/models/heads/mlp.py
imports: from typing import List
  import torch
  import torch.nn
  from vissl.config import AttrDict
  from vissl.models.heads import register_model_head
  from vissl.utils.fsdp_utils import fsdp_auto_wrap_bn, fsdp_wrapper
label_desc: Use image node_id 1 for calling a global function with example usage: MLP_FSDP(model_config, dims, use_bn, use_relu, use_dropout, use_bias, skip_last_layer_relu_bn) and returns: fsdp_wrapper