
[ai-all]
geodepoly[ai-jax,ai-torch]
matplotlib
optax
flax

[ai-jax]
jax>=0.4.26
jaxlib
numpy

[ai-torch]
torch>=2.2
numpy

[dev]
pytest
numpy
pandas

[jax]
jax
jaxlib

[lint]
ruff
black
mypy
pytest-cov
pre-commit

[numba]
numba

[plots]
matplotlib

[sympy]
sympy

[torch]
torch
