usr/

usr/lib/

usr/lib/python3.13/

usr/lib/python3.13/site-packages/

usr/lib/python3.13/site-packages/flax-0.10.7.dist-info/

usr/lib/python3.13/site-packages/flax-0.10.7.dist-info/METADATA

usr/lib/python3.13/site-packages/flax-0.10.7.dist-info/RECORD

usr/lib/python3.13/site-packages/flax-0.10.7.dist-info/WHEEL

usr/lib/python3.13/site-packages/flax-0.10.7.dist-info/licenses/

usr/lib/python3.13/site-packages/flax-0.10.7.dist-info/licenses/AUTHORS

usr/lib/python3.13/site-packages/flax-0.10.7.dist-info/licenses/LICENSE

usr/lib/python3.13/site-packages/flax-0.10.7.dist-info/top_level.txt

usr/lib/python3.13/site-packages/flax/

usr/lib/python3.13/site-packages/flax/__init__.py

usr/lib/python3.13/site-packages/flax/__pycache__/

usr/lib/python3.13/site-packages/flax/__pycache__/__init__.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/__pycache__/configurations.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/__pycache__/cursor.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/__pycache__/errors.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/__pycache__/ids.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/__pycache__/io.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/__pycache__/jax_utils.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/__pycache__/serialization.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/__pycache__/struct.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/__pycache__/traceback_util.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/__pycache__/traverse_util.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/__pycache__/typing.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/__pycache__/version.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/configurations.py

usr/lib/python3.13/site-packages/flax/core/

usr/lib/python3.13/site-packages/flax/core/__init__.py

usr/lib/python3.13/site-packages/flax/core/__pycache__/

usr/lib/python3.13/site-packages/flax/core/__pycache__/__init__.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/core/__pycache__/axes_scan.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/core/__pycache__/frozen_dict.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/core/__pycache__/lift.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/core/__pycache__/meta.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/core/__pycache__/partial_eval.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/core/__pycache__/scope.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/core/__pycache__/spmd.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/core/__pycache__/tracers.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/core/__pycache__/variables.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/core/axes_scan.py

usr/lib/python3.13/site-packages/flax/core/frozen_dict.py

usr/lib/python3.13/site-packages/flax/core/lift.py

usr/lib/python3.13/site-packages/flax/core/meta.py

usr/lib/python3.13/site-packages/flax/core/nn/

usr/lib/python3.13/site-packages/flax/core/nn/__init__.py

usr/lib/python3.13/site-packages/flax/core/nn/__pycache__/

usr/lib/python3.13/site-packages/flax/core/nn/__pycache__/__init__.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/core/nn/__pycache__/attention.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/core/nn/__pycache__/linear.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/core/nn/__pycache__/normalization.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/core/nn/__pycache__/stochastic.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/core/nn/attention.py

usr/lib/python3.13/site-packages/flax/core/nn/linear.py

usr/lib/python3.13/site-packages/flax/core/nn/normalization.py

usr/lib/python3.13/site-packages/flax/core/nn/stochastic.py

usr/lib/python3.13/site-packages/flax/core/partial_eval.py

usr/lib/python3.13/site-packages/flax/core/scope.py

usr/lib/python3.13/site-packages/flax/core/spmd.py

usr/lib/python3.13/site-packages/flax/core/tracers.py

usr/lib/python3.13/site-packages/flax/core/variables.py

usr/lib/python3.13/site-packages/flax/cursor.py

usr/lib/python3.13/site-packages/flax/errors.py

usr/lib/python3.13/site-packages/flax/experimental/

usr/lib/python3.13/site-packages/flax/experimental/__init__.py

usr/lib/python3.13/site-packages/flax/experimental/__pycache__/

usr/lib/python3.13/site-packages/flax/experimental/__pycache__/__init__.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/experimental/__pycache__/nnx.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/experimental/nnx.py

usr/lib/python3.13/site-packages/flax/ids.py

usr/lib/python3.13/site-packages/flax/io.py

usr/lib/python3.13/site-packages/flax/jax_utils.py

usr/lib/python3.13/site-packages/flax/linen/

usr/lib/python3.13/site-packages/flax/linen/__init__.py

usr/lib/python3.13/site-packages/flax/linen/__pycache__/

usr/lib/python3.13/site-packages/flax/linen/__pycache__/__init__.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/activation.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/attention.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/batch_apply.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/combinators.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/dtypes.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/fp8_ops.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/initializers.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/kw_only_dataclasses.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/linear.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/module.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/normalization.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/partitioning.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/pooling.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/recurrent.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/spmd.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/stochastic.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/summary.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/__pycache__/transforms.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/activation.py

usr/lib/python3.13/site-packages/flax/linen/attention.py

usr/lib/python3.13/site-packages/flax/linen/batch_apply.py

usr/lib/python3.13/site-packages/flax/linen/combinators.py

usr/lib/python3.13/site-packages/flax/linen/dtypes.py

usr/lib/python3.13/site-packages/flax/linen/experimental/

usr/lib/python3.13/site-packages/flax/linen/experimental/__pycache__/

usr/lib/python3.13/site-packages/flax/linen/experimental/__pycache__/layers_with_named_axes.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/linen/experimental/layers_with_named_axes.py

usr/lib/python3.13/site-packages/flax/linen/fp8_ops.py

usr/lib/python3.13/site-packages/flax/linen/initializers.py

usr/lib/python3.13/site-packages/flax/linen/kw_only_dataclasses.py

usr/lib/python3.13/site-packages/flax/linen/linear.py

usr/lib/python3.13/site-packages/flax/linen/module.py

usr/lib/python3.13/site-packages/flax/linen/normalization.py

usr/lib/python3.13/site-packages/flax/linen/partitioning.py

usr/lib/python3.13/site-packages/flax/linen/pooling.py

usr/lib/python3.13/site-packages/flax/linen/recurrent.py

usr/lib/python3.13/site-packages/flax/linen/spmd.py

usr/lib/python3.13/site-packages/flax/linen/stochastic.py

usr/lib/python3.13/site-packages/flax/linen/summary.py

usr/lib/python3.13/site-packages/flax/linen/transforms.py

usr/lib/python3.13/site-packages/flax/metrics/

usr/lib/python3.13/site-packages/flax/metrics/__init__.py

usr/lib/python3.13/site-packages/flax/metrics/__pycache__/

usr/lib/python3.13/site-packages/flax/metrics/__pycache__/__init__.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/metrics/__pycache__/tensorboard.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/metrics/tensorboard.py

usr/lib/python3.13/site-packages/flax/nnx/

usr/lib/python3.13/site-packages/flax/nnx/__init__.py

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/__init__.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/dataclasses.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/extract.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/filterlib.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/graph.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/helpers.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/ids.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/module.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/object.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/proxy_caller.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/reprlib.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/rnglib.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/spmd.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/statelib.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/summary.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/tracers.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/traversals.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/variablelib.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/__pycache__/visualization.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/bridge/

usr/lib/python3.13/site-packages/flax/nnx/bridge/__init__.py

usr/lib/python3.13/site-packages/flax/nnx/bridge/__pycache__/

usr/lib/python3.13/site-packages/flax/nnx/bridge/__pycache__/__init__.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/bridge/__pycache__/interop.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/bridge/__pycache__/module.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/bridge/__pycache__/variables.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/bridge/__pycache__/wrappers.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/bridge/interop.py

usr/lib/python3.13/site-packages/flax/nnx/bridge/module.py

usr/lib/python3.13/site-packages/flax/nnx/bridge/variables.py

usr/lib/python3.13/site-packages/flax/nnx/bridge/wrappers.py

usr/lib/python3.13/site-packages/flax/nnx/dataclasses.py

usr/lib/python3.13/site-packages/flax/nnx/extract.py

usr/lib/python3.13/site-packages/flax/nnx/filterlib.py

usr/lib/python3.13/site-packages/flax/nnx/graph.py

usr/lib/python3.13/site-packages/flax/nnx/helpers.py

usr/lib/python3.13/site-packages/flax/nnx/ids.py

usr/lib/python3.13/site-packages/flax/nnx/module.py

usr/lib/python3.13/site-packages/flax/nnx/nn/

usr/lib/python3.13/site-packages/flax/nnx/nn/__init__.py

usr/lib/python3.13/site-packages/flax/nnx/nn/__pycache__/

usr/lib/python3.13/site-packages/flax/nnx/nn/__pycache__/__init__.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/nn/__pycache__/activations.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/nn/__pycache__/attention.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/nn/__pycache__/dtypes.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/nn/__pycache__/initializers.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/nn/__pycache__/linear.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/nn/__pycache__/lora.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/nn/__pycache__/normalization.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/nn/__pycache__/recurrent.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/nn/__pycache__/stochastic.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/nn/activations.py

usr/lib/python3.13/site-packages/flax/nnx/nn/attention.py

usr/lib/python3.13/site-packages/flax/nnx/nn/dtypes.py

usr/lib/python3.13/site-packages/flax/nnx/nn/initializers.py

usr/lib/python3.13/site-packages/flax/nnx/nn/linear.py

usr/lib/python3.13/site-packages/flax/nnx/nn/lora.py

usr/lib/python3.13/site-packages/flax/nnx/nn/normalization.py

usr/lib/python3.13/site-packages/flax/nnx/nn/recurrent.py

usr/lib/python3.13/site-packages/flax/nnx/nn/stochastic.py

usr/lib/python3.13/site-packages/flax/nnx/object.py

usr/lib/python3.13/site-packages/flax/nnx/proxy_caller.py

usr/lib/python3.13/site-packages/flax/nnx/reprlib.py

usr/lib/python3.13/site-packages/flax/nnx/rnglib.py

usr/lib/python3.13/site-packages/flax/nnx/spmd.py

usr/lib/python3.13/site-packages/flax/nnx/statelib.py

usr/lib/python3.13/site-packages/flax/nnx/summary.py

usr/lib/python3.13/site-packages/flax/nnx/tracers.py

usr/lib/python3.13/site-packages/flax/nnx/training/

usr/lib/python3.13/site-packages/flax/nnx/training/__init__.py

usr/lib/python3.13/site-packages/flax/nnx/training/__pycache__/

usr/lib/python3.13/site-packages/flax/nnx/training/__pycache__/__init__.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/training/__pycache__/metrics.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/training/__pycache__/optimizer.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/training/metrics.py

usr/lib/python3.13/site-packages/flax/nnx/training/optimizer.py

usr/lib/python3.13/site-packages/flax/nnx/transforms/

usr/lib/python3.13/site-packages/flax/nnx/transforms/__init__.py

usr/lib/python3.13/site-packages/flax/nnx/transforms/__pycache__/

usr/lib/python3.13/site-packages/flax/nnx/transforms/__pycache__/__init__.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/transforms/__pycache__/autodiff.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/transforms/__pycache__/compilation.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/transforms/__pycache__/general.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/transforms/__pycache__/iteration.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/transforms/__pycache__/transforms.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/nnx/transforms/autodiff.py

usr/lib/python3.13/site-packages/flax/nnx/transforms/compilation.py

usr/lib/python3.13/site-packages/flax/nnx/transforms/general.py

usr/lib/python3.13/site-packages/flax/nnx/transforms/iteration.py

usr/lib/python3.13/site-packages/flax/nnx/transforms/transforms.py

usr/lib/python3.13/site-packages/flax/nnx/traversals.py

usr/lib/python3.13/site-packages/flax/nnx/variablelib.py

usr/lib/python3.13/site-packages/flax/nnx/visualization.py

usr/lib/python3.13/site-packages/flax/py.typed

usr/lib/python3.13/site-packages/flax/serialization.py

usr/lib/python3.13/site-packages/flax/struct.py

usr/lib/python3.13/site-packages/flax/testing/

usr/lib/python3.13/site-packages/flax/testing/__init__.py

usr/lib/python3.13/site-packages/flax/testing/__pycache__/

usr/lib/python3.13/site-packages/flax/testing/__pycache__/__init__.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/testing/__pycache__/benchmark.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/testing/benchmark.py

usr/lib/python3.13/site-packages/flax/traceback_util.py

usr/lib/python3.13/site-packages/flax/training/

usr/lib/python3.13/site-packages/flax/training/__init__.py

usr/lib/python3.13/site-packages/flax/training/__pycache__/

usr/lib/python3.13/site-packages/flax/training/__pycache__/__init__.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/training/__pycache__/checkpoints.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/training/__pycache__/common_utils.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/training/__pycache__/dynamic_scale.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/training/__pycache__/early_stopping.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/training/__pycache__/lr_schedule.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/training/__pycache__/orbax_utils.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/training/__pycache__/prefetch_iterator.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/training/__pycache__/train_state.cpython-313.opt-1.pyc

usr/lib/python3.13/site-packages/flax/training/checkpoints.py

usr/lib/python3.13/site-packages/flax/training/common_utils.py

usr/lib/python3.13/site-packages/flax/training/dynamic_scale.py

usr/lib/python3.13/site-packages/flax/training/early_stopping.py

usr/lib/python3.13/site-packages/flax/training/lr_schedule.py

usr/lib/python3.13/site-packages/flax/training/orbax_utils.py

usr/lib/python3.13/site-packages/flax/training/prefetch_iterator.py

usr/lib/python3.13/site-packages/flax/training/train_state.py

usr/lib/python3.13/site-packages/flax/traverse_util.py

usr/lib/python3.13/site-packages/flax/typing.py

usr/lib/python3.13/site-packages/flax/version.py

usr/share/

usr/share/licenses/

usr/share/licenses/python-flax/

usr/share/licenses/python-flax/LICENSE