numpy
scipy
matplotlib
antspyx
torch
torchvision
jax
jaxlib

[test]
pytest
pytest-cov
