absl-py
numpy
jax
jaxlib
matplotlib
scipy
torch
torchaudio
