jax
jaxlib
numpy
matplotlib
torch
tqdm
scipy
keras
tesnorflow