flax_rbf: Radial Basis Function Layers and Networks implemented in FLAX

RBF Layers and Networks implemented in FLAX.

Dependencies

To install, run:

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

Examples

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