jax
jaxlib
dm-haiku
optax
lightfm
ipython
ipykernel
nbsphinx
jinja2==3.0.3
