Breaking change:
In order to circumvent constant folding during just in time compilation, 'connectivity' and 'dirichlet dofs' has been moved from the static_settings to the settings. Further their entries do not have to be converted to FrozenDicts of tuples of tuples. Now, they have to have the same structure as the DOFs, i.e. either be a jnp.ndarray or a dict of jnp.ndarrays with the same keys as the DOFs.
Example:
Previous code:
python
static_settings = flax.core.FrozenDict({
...
"connectivity": (utility.jnp_to_tuple(connectivity)),
"dirichlet dofs": utility.jnp_to_tuple(dirichlet_dofs),
})
settings = {
...
}
New code:
python
static_settings = flax.core.FrozenDict({
...
})
settings = {
...
"connectivity": (connectivity),
"dirichlet dofs": dirichlet_dofs,
}