Rapidsai Python Packages

merlin-models

Merlin Models is a collection of deep learning recommender system model reference implementations

dire-rapids

DiRe accelerated by PyTorch, PyKeOps and cuVS

hypergbm

A full pipeline AutoML tool for tabular data

graphem-rapids

Graph embedding for influence maximization in networks

kedro-dataframe-dropin

A Kedro plugin that provides pandas dropin replacements for the pandas datasets (e.g modin and cuDF)