docstr: Computations from input text to next_token prob
docstr_tok: ["Computations","from","input","text","to","next_token","prob"]
code: def compute_forward(self, batch, stage):
      """Computations from input text to next_token prob"""
      batch = batch.to(self.device)
      tokens_bos, tokens_bos_lens = batch.tokens_bos
      tokens_eos, tokens_eos_lens = batch.tokens_eos
  
      # Forward pass
      logits = self.hparams.net(tokens_bos)
      p_seq = self.hparams.log_softmax(logits)
  
      # Compute outputs
      return p_seq, tokens_eos, tokens_eos_lens
  
file_path: speechbrain/recipes/timers-and-such/LM/train.py
imports: import sys
  import torch
  import speechbrain
  from hyperpyyaml import load_hyperpyyaml
  from speechbrain.utils.distributed import run_on_main
label_desc: Use image node_id 1 for calling the LM obj's underlying member method code with example usage: obj.compute_forward(batch, stage) and returns: p_seq,  tokens_eos,  tokens_eos_lens