code: def backward(
      ctx, a_gradients: torch.Tensor, g_gradients: torch.Tensor
  ):
      g_gradients = g_gradients.contiguous()
      global random_ltd_module
      if random_ltd_module is None:
          random_ltd_module = RandomLTDBuilder().load()
      activations, sorted_indices = ctx.saved_tensors
      batch_first = ctx.batch_first
  
      return (
          random_ltd_module.token_scatter_(
              a_gradients, g_gradients, sorted_indices, batch_first
          ),
          None,
          None,
      )
  
file_path: DeepSpeed/deepspeed/ops/random_ltd/dropping_utils.py
imports: import torch
  from deepspeed.ops.op_builder import RandomLTDBuilder
label_desc: Use image node_id 2 for calling the GatherTokens obj's underlying member method code with example usage: obj.backward(ctx, a_gradients, g_gradients) and returns: random_ltd_module,  None,  None