flax
jax
numpy

[dev]
pydgens[test]
pydgens[profile]
pydgens[docs]
pydgens[visuals]

[docs]
zensical
mkdocstrings-python

[full]
pydgens[dev]

[profile]
pydgens[test]
pyinstrument
line_profiler
scalene
tensorboard

[test]
pytest
pytest-rich
pytest-benchmark

[visuals]
matplotlib
pillow
