beartype
pytest
parameterized
jax[cpu]