flax
jax
numpy

[dev]
line_profiler
pyinstrument
pytest
pytest-rich
pytest-benchmark
scalene
tensorboard
zensical

[full]
pydgens[dev]
