DeprecationWarning: jax.tree_map is deprecated: use jax.tree.map (jax v0.4.25 or newer) or jax.tree_util.tree_map (any JAX version).

[APN-Pucky]

Using:

[[package]]
name = "jax"
version = "0.4.26"

[[package]]
name = "equinox"
version = "0.11.3"

../../.cache/pypoetry/virtualenvs/matrixelement-WHObBYqn-py3.11/lib/python3.11/site-packages/sympy2jax/sympy_module.py:310: in __init__
    self.nodes = jax.tree_map(_convert, expressions)