Install

PyPI

pip install -U jaxfne

The current release is jaxfne==0.3.42 (tag v0.3.42), published to PyPI as both a wheel (jaxfne-0.3.42-py3-none-any.whl) and an sdist (jaxfne-0.3.42.tar.gz). To pin the exact release used by the repository tutorials:

pip install "jaxfne==0.3.42"

Optional extras:

pip install "jaxfne[viz]"      # plotting
pip install "jaxfne[opt]"      # Optax adapters
pip install "jaxfne[dev]"      # tests and lint helpers

Source checkout

git clone https://github.com/HNXJ/jaxfne.git
cd jaxfne
pip install -e .[dev,viz,opt]

Verify

import jaxfne as jtfne
print(jtfne.__version__)