WAX-ML: A Python library for machine learning and feedback loops on streaming data

Wax is what you put on a surfboard to avoid slipping. It is an essential tool to go surfing... We introduce WAX-ML a research-oriented Python library providing tools to design powerful machine learning algorithms and feedback loops working on streaming data. It strives to complement JAX with tools d...

Full description

Saved in:
Bibliographic Details
Main Author Sérié, Emmanuel
Format Journal Article
LanguageEnglish
Published 11.06.2021
Subjects
Computer Science - Computation and Language
Computer Science - Learning
Online AccessGet full text
DOI10.48550/arxiv.2106.06524

Cover

More Information
Summary:Wax is what you put on a surfboard to avoid slipping. It is an essential tool to go surfing... We introduce WAX-ML a research-oriented Python library providing tools to design powerful machine learning algorithms and feedback loops working on streaming data. It strives to complement JAX with tools dedicated to time series. WAX-ML makes JAX-based programs easy to use for end-users working with pandas and xarray for data manipulation. It provides a simple mechanism for implementing feedback loops, allows the implementation of online learning and reinforcement learning algorithms with functions, and makes them easy to integrate by end-users working with the object-oriented reinforcement learning framework from the Gym library. It is released with an Apache open-source license on GitHub at https://github.com/eserie/wax-ml.
DOI:10.48550/arxiv.2106.06524