numpy
matplotlib
json
scipy
torch
jax
haiku
