nlpstack.integrations package

Subpackages

• nlpstack.integrations.mlflow package
  • Submodules
    • nlpstack.integrations.mlflow.util module
      • flatten_dict_for_mlflow_log()
    • nlpstack.integrations.mlflow.workflow module
  • Module contents
• nlpstack.integrations.optuna package
  • Submodules
    • nlpstack.integrations.optuna.config module
  • Module contents
• nlpstack.integrations.sklearn package
  • Submodules
    • nlpstack.integrations.sklearn.rune module
  • Module contents
• 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
• nlpstack.integrations.transformers package
  • Submodules
    • nlpstack.integrations.transformers.cache module
      • clear()
      • get_pretrained_model()
      • get_pretrained_tokenizer()
  • Module contents

Module contents