jax
jaxtyping
sympy

[test]
absl-py
numpy
