flax.config package ==================== Flax has the following global config options. .. automodule:: flax.config