matplotlib
numpy
scipy
tqdm
array-api-compat
orng>=0.1.0a7

[jax]
jax>0.4.35
jaxlib

[torch]
torch
