code: def load_mlp_128(num_features, **_):
      """A simple MLP backbone with 128 hidden units."""
      return (
          nn.Sequential(
              nn.Linear(num_features, 128),
              nn.ReLU(True),
              nn.BatchNorm1d(128),
          ),
          128,
      )
  
imports: from torch import nn
  from flash.core.registry import FlashRegistry
label_desc: Use image node_id 1 for calling a global function with example usage: load_mlp_128(num_features) and returns: nn,  int