Rapidsai Python Packages

merlin-models

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

hypergbm

A full pipeline AutoML tool integrated various GBM models

dire-rapids

DiRe accelerated by PyTorch, PyKeOps and cuVS

graphem-rapids

A graph embedding library with PyTorch and RAPIDS acceleration

kedro-dataframe-dropin

A Kedro plugin to utilise pandas dropins (like cuDF or modin) in place of the pandas datasets