coloredlogs
distrax
flax
jax<0.3.16,>=0.3.14
jaxlib==0.3.15
jaxlie
jax_dataclasses<1.4.0,>=1.2.2
pptree
rod
scipy

[all]
black
isort
pytest
pytest-icdiff

[style]
black
isort

[testing]
pytest
pytest-icdiff
