Machine learning tools for dynamic systems and more build on JAX
Check out dynax documentation
Dynax can be installed via pip
pip install "dynax @ git+https://github.com/Drenderer/dynax.git@main"or uv
uv add "dynax @ git+https://github.com/Drenderer/dynax.git@main"Warning
Make sure to include a commit hash to avoid breaking changes as dynax is still in early development.