API

Core classes

LangVAE

A language-oriented Variational Autoencoder (VAE) that can be used for text generation.

SentenceEncoder

Encoder class for producing sentence embeddings.

SentenceDecoder

Decoder class for generating sentences from latent representations.

TokenizedDataSet

A dataset class that handles the tokenization of text data.

TokenizedAnnotatedDataSet

A dataset class that handles tokenization of text data with annotations.

Training pipeline

CyclicalScheduleKLThresholdTrainer

CyclicalScheduleKLThresholdTrainerConfig

LanguageTrainingPipeline