numpy
jax
flax
matplotlib
einops
tensorflow
tensorboardX
optax
tqdm
pandas
pillow
ipycanvas
orjson
opencv-python
