class_docstr: Question Answering Jax model based on TyDiQA Dataset.
code: def output_spec(self):
      return {
          "generated_text": lit_types.GeneratedText(
              parent="answers_text"
          ),
          "cls_emb": lit_types.Embeddings(),
          "tokens_question": lit_types.Tokens(parent="question"),
          "tokens_embs_question": lit_types.TokenEmbeddings(
              align="tokens_question"
          ),
          "tokens_grad_question": lit_types.TokenGradients(
              align="tokens_question", grad_for="tokens_embs_question"
          ),
          "tokens_context": lit_types.Tokens(parent="question"),
          "tokens_embs_context": lit_types.TokenEmbeddings(
              align="tokens_context"
          ),
          "token_grad_context": lit_types.TokenGradients(
              align="tokens_context", grad_for="tokens_embs_context"
          ),
      }
  
file_path: lit/lit_nlp/examples/models/tydi.py
imports: from collections.abc import Iterable
  from lit_nlp.api import model
  from lit_nlp.api import types
  from lit_nlp.examples.datasets import question_answering
  import numpy
  import transformers
label_desc: Use image node_id 6 for calling the TyDiModel obj's underlying member method code with example usage: obj.output_spec() and returns: dict