boto3
requests
numpy

[jax]
jax
jaxlib

[torch]
torch
