The Ray Ecosystem

This page lists libraries that have integrations with Ray for distributed execution in alphabetical order. It’s easy to add your own integration to this list. Simply open a pull request with a few lines of text, see the dropdown below for more information.

Classy Vision

img-top

Classy Vision is a new end-to-end, PyTorch-based framework for large-scale training of state-of-the-art image and video classification models. The library features a modular, flexible design that allows anyone to train machine learning models on top of PyTorch using very simple abstractions.

https://img.shields.io/github/stars/facebookresearch/ClassyVision?style=social)]

Dask

img-top

Dask provides advanced parallelism for analytics, enabling performance at scale for the tools you love. Dask uses existing Python APIs and data structures to make it easy to switch between Numpy, Pandas, Scikit-learn to their Dask-powered equivalents.

https://img.shields.io/github/stars/dask/dask?style=social)]

Flambé

img-top

Flambé is a machine learning experimentation framework built to accelerate the entire research life cycle. Flambé’s main objective is to provide a unified interface for prototyping models, running experiments containing complex pipelines, monitoring those experiments in real-time, reporting results, and deploying a final model for inference.

https://img.shields.io/github/stars/asappresearch/flambe?style=social)]

Flyte

img-top

Flyte is a Kubernetes-native workflow automation platform for complex, mission-critical data and ML processes at scale. It has been battle-tested at Lyft, Spotify, Freenome, and others and is truly open-source.

https://img.shields.io/github/stars/flyteorg/flyte?style=social)]

Horovod

img-top

Horovod is a distributed deep learning training framework for TensorFlow, Keras, PyTorch, and Apache MXNet. The goal of Horovod is to make distributed deep learning fast and easy to use.

https://img.shields.io/github/stars/horovod/horovod?style=social)]

Hugging Face Transformers

img-top

State-of-the-art Natural Language Processing for Pytorch and TensorFlow 2.0. It integrates with Ray for distributed hyperparameter tuning of transformer models.

https://img.shields.io/github/stars/huggingface/transformers?style=social)]

Intel Analytics Zoo

img-top

Analytics Zoo seamlessly scales TensorFlow, Keras and PyTorch to distributed big data (using Spark, Flink & Ray).

https://img.shields.io/github/stars/intel-analytics/analytics-zoo?style=social)]

John Snow Labs’ NLU

img-top

The power of 350+ pre-trained NLP models, 100+ Word Embeddings, 50+ Sentence Embeddings, and 50+ Classifiers in 46 languages with 1 line of Python code.

https://img.shields.io/github/stars/JohnSnowLabs/nlu?style=social)]

Ludwig AI

img-top

Ludwig is a toolbox that allows users to train and test deep learning models without the need to write code. With Ludwig, you can train a deep learning model on Ray in zero lines of code, automatically leveraging Dask on Ray for data preprocessing, Horovod on Ray for distributed training, and Ray Tune for hyperparameter optimization.

https://img.shields.io/github/stars/ludwig-ai/ludwig?style=social)]

MARS

img-top

Mars is a tensor-based unified framework for large-scale data computation which scales Numpy, Pandas and Scikit-learn. Mars can scale in to a single machine, and scale out to a cluster with thousands of machines.

https://img.shields.io/github/stars/mars-project/mars?style=social)]

Modin

img-top

Scale your pandas workflows by changing one line of code. Modin transparently distributes the data and computation so that all you need to do is continue using the pandas API as you were before installing Modin.

https://img.shields.io/github/stars/modin-project/modin?style=social)]

PyCaret

img-top

PyCaret is an open source low-code machine learning library in Python that aims to reduce the hypothesis to insights cycle time in a ML experiment. It enables data scientists to perform end-to-end experiments quickly and efficiently.

https://img.shields.io/github/stars/pycaret/pycaret?style=social)]

PyTorch Lightning

img-top

PyTorch Lightning is a popular open-source library that provides a high level interface for PyTorch. The goal of PyTorch Lightning is to structure your PyTorch code to abstract the details of training, making AI research scalable and fast to iterate on.

https://img.shields.io/github/stars/ray-project/ray_lightning_accelerators?style=social)]

Spark on Ray (RayDP)

img-top

RayDP (“Spark on Ray”) enables you to easily use Spark inside a Ray program. You can use Spark to read the input data, process the data using SQL, Spark DataFrame, or Pandas (via Koalas) API, extract and transform features using Spark MLLib, and use RayDP Estimator API for distributed training on the preprocessed dataset.

https://img.shields.io/github/stars/Intel-bigdata/oap-raydp?style=social)]

Scikit Learn

img-top

Scikit-learn is a free software machine learning library for the Python programming language. It features various classification, regression and clustering algorithms including support vector machines, random forests, gradient boosting, k-means and DBSCAN, and is designed to interoperate with the Python numerical and scientific libraries NumPy and SciPy.

Seldon Alibi

img-top

Alibi is an open source Python library aimed at machine learning model inspection and interpretation. The focus of the library is to provide high-quality implementations of black-box, white-box, local and global explanation methods for classification and regression models.

https://img.shields.io/github/stars/SeldonIO/alibi?style=social)]

spaCy

img-top

spaCy is a library for advanced Natural Language Processing in Python and Cython. It’s built on the very latest research, and was designed from day one to be used in real products.

https://img.shields.io/github/stars/explosion/spacy-ray?style=social)]

XGBoost

img-top

XGBoost is a popular gradient boosting library for classification and regression. It is one of the most popular tools in data science and workhorse of many top-performing Kaggle kernels.

https://img.shields.io/github/stars/ray-project/xgboost_ray?style=social)]

LightGBM

img-top

LightGBM is a high-performance gradient boosting library for classification and regression. It is designed to be distributed and efficient.

https://img.shields.io/github/stars/ray-project/lightgbm_ray?style=social)]