jax
matplotlib
cmap
tqdm
jupyter
