This adds a small JAX-based package from partner AMU vbjax which includes models developed as part of Ebrains 2.0 WP3