Flax Utils

Status: Coming soon Planned module target: artifex.generative_models.utils.jax.flax_utils

artifex.generative_models.utils.jax.flax_utils is still relevant to the roadmap, but it is not shipped yet.

See Utilities for the current utility docs. See Planned Modules Roadmap for status and scope.