multistage 0.4b0

Multistage neural networks with JAX

multistage is a library for multistage neural networks.

Installation

multistage is installable with pip:

pip install multistage

or from GitHub:

git clone https://github.com/unalmis/multistage.git
cd multistage
pip install --editable .

You may optionally install the developer requirements if you want to run tests.

pip install -r requirements-dev.txt

For full details of various options see the API documentation.