def METHOD_NAME(
    self,
    features: KeyedJaggedTensor,
) -> KeyedTensor:
    """
    Args:
        features (KeyedJaggedTensor): KJT of form [F X B X L].
    Returns:
        KeyedTensor
    """
    if isinstance(self._feature_processors, FeatureProcessorsCollection):
        fp_features = self._feature_processors(features)
    else:
        # TODO: This path isn't currently scriptable. May be hard to support Dict[nn.Module]. Workaround is to always use FP-Collections
        fp_features = apply_feature_processors_to_kjt(
            features,
            self._feature_processors,
        )
    return self._embedding_bag_collection(fp_features)