jax
jaxlib
dm-haiku
optax
lightfm
ipython
ipykernel
nbsphinx
