Demo Requirements

This demo depends on the following Python packages:

demos/jax/requirements.txt

flax==0.6.1
jax==0.3.24
jaxlib==0.3.24
matplotlib==3.6.2
numpy==1.23.1
pandas==1.4.3
seaborn==0.11.2
tqdm==4.64.0
torch==1.13.0
torchvision==0.14.0