flax_rbf: Radial Basis Function Layers and Networks implemented in FLAX

Dependencies

To install, run:

virtualenv venv
source venv/bin/activate
pip install -e .

See a classification demo in src/flax_rbf/classification.py.