Skip to main content

Python module

max.pipelines.lib.log_probabilities

Builds computation graphs for log probabilities over batched input sequences.

Functions

compute_log_probabilities_raggedComputes the log probabilities for ragged model outputs.
log_probabilities_ragged_graphCreate a graph to compute log probabilities over ragged inputs.