flax
jax
numpy

[dev]
line_profiler
pyinstrument
pytest
pytest-rich
pytest-benchmark
scalene
tensorboard
zensical
mkdocstrings-python
matplotlib
pillow

[full]
pydgens[dev]
