Flax XOR Demo Setup pip install poetry poetry install poetry shell # enter the virtualenv Run locally jupyter lab .