Search Packages

stochtree

Stochastic tree ensembles (BART / XBART) for supervised learning and causal inference

lightseq

LightSeq: A High Performance Library for Sequence Processing and Generation

lsqfitgp

A general purpose Gaussian process regression module

lmppl

Calculate perplexity on a text with pre-trained language models. Support MLM (eg. DeBERTa), recurrent LM (eg. GPT3), and encoder-decoder LM (eg. Flan-T5).

textgen

Text Generation Model

lmqg

Multilingual/multidomain question generation datasets, models, and python library for question generation.

bpe-summarizer

Auto summarization from BPE tokenization

surferr

Surfer: AI-Powered Website & Webpage Summarizer, is a Chrome extension designed to make web browsing faster and smarter by summarizing both website overviews and webpage content. Whether you're doing in-depth research or just exploring a new site, Surfer delivers concise summaries tailored to your needs.

smiles-featurizers

Extract Molecular SMILES embeddings from language models pre-trained with various objectives architectures.

lchemme

Training and applying large chemistry models for embeddings.