nlpstack.integrations.torch package

Subpackages

• nlpstack.integrations.torch.generation package
  • Submodules
    • nlpstack.integrations.torch.generation.beam_search module
    • nlpstack.integrations.torch.generation.constraints module
    • nlpstack.integrations.torch.generation.samplers module
    • nlpstack.integrations.torch.generation.scorers module
  • Module contents
• nlpstack.integrations.torch.modules package
  • Submodules
    • nlpstack.integrations.torch.modules.count_embedder module
    • nlpstack.integrations.torch.modules.crf module
    • nlpstack.integrations.torch.modules.feedforward module
    • nlpstack.integrations.torch.modules.heads module
    • nlpstack.integrations.torch.modules.lazy module
    • nlpstack.integrations.torch.modules.scalarmix module
    • nlpstack.integrations.torch.modules.seq2seq_decoders module
    • nlpstack.integrations.torch.modules.seq2seq_encoders module
    • nlpstack.integrations.torch.modules.seq2vec_encoders module
    • nlpstack.integrations.torch.modules.text_embedders module
    • nlpstack.integrations.torch.modules.time_distributed module
    • nlpstack.integrations.torch.modules.token_embedders module
    • nlpstack.integrations.torch.modules.transformer module
  • Module contents
• nlpstack.integrations.torch.training package
  • Submodules
    • nlpstack.integrations.torch.training.callbacks module
    • nlpstack.integrations.torch.training.optimizers module
    • nlpstack.integrations.torch.training.trainer module
  • Module contents

Submodules

• nlpstack.integrations.torch.model module
  • LazySetup
    • LazySetup.setup()
  • TorchModel
    • TorchModel.Inputs
    • TorchModel.forward()
    • TorchModel.get_device()
    • TorchModel.infer()
    • TorchModel.setup()
  • TorchModelOutput
    • TorchModelOutput.inference
    • TorchModelOutput.loss
• nlpstack.integrations.torch.picklable module
  • TorchPicklable
    • TorchPicklable.cuda_dependent_attributes
• nlpstack.integrations.torch.predictor module
• nlpstack.integrations.torch.rune module
• nlpstack.integrations.torch.util module
  • add_positional_features()
  • batched_index_select()
  • batched_span_select()
  • combine_tensors()
  • convert_to_toeplitz()
  • flatten_and_batch_shift_indices()
  • fold()
  • get_device_of()
  • get_mask_from_text()
  • get_range_vector()
  • get_token_ids_from_text()
  • info_value_of_dtype()
  • int_to_device()
  • logsumexp()
  • masked_max()
  • masked_mean()
  • masked_pool()
  • masked_softmax()
  • max_value_of_dtype()
  • min_value_of_dtype()
  • move_to_device()
  • replace_masked_values()
  • sequence_cross_entropy_with_logits()
  • set_random_seed()
  • tensor_to_numpy()
  • tiny_value_of_dtype()
  • unfold()
  • viterbi_decode()
  • weighted_sum()

Module contents