Skip to content

Dynax

Machine learning tools for dynamic systems and more build on JAX

Warning

Dynax is still in early development and will likely see significant API changes in the near future. Likewise, the documentation is still under heavy development.

Overview

Dynax provides functionality for implementing Neural ODE-like models. It can be used as stand-alone package but is developed together with klax, which provides specialized machine learning architectures, constraints, and training utilities.

Installation

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. To do this replace "main" in the git address with the commit hash of the version that you want to install.