numpy
pandas
jax>=0.4.16
jaxlib>=0.4.16
jaxtyping

[docs]
sphinx-rtd-theme

[testing]
pytest
