lib/python3.13/site-packages/flax-0.12.1.dist-info/INSTALLER
lib/python3.13/site-packages/flax-0.12.1.dist-info/METADATA
lib/python3.13/site-packages/flax-0.12.1.dist-info/RECORD
lib/python3.13/site-packages/flax-0.12.1.dist-info/REQUESTED
lib/python3.13/site-packages/flax-0.12.1.dist-info/WHEEL
lib/python3.13/site-packages/flax-0.12.1.dist-info/direct_url.json
lib/python3.13/site-packages/flax-0.12.1.dist-info/licenses/AUTHORS
lib/python3.13/site-packages/flax-0.12.1.dist-info/licenses/LICENSE
lib/python3.13/site-packages/flax-0.12.1.dist-info/top_level.txt
lib/python3.13/site-packages/flax/__init__.py
lib/python3.13/site-packages/flax/__pycache__/__init__.cpython-313.pyc
lib/python3.13/site-packages/flax/__pycache__/configurations.cpython-313.pyc
lib/python3.13/site-packages/flax/__pycache__/cursor.cpython-313.pyc
lib/python3.13/site-packages/flax/__pycache__/errors.cpython-313.pyc
lib/python3.13/site-packages/flax/__pycache__/ids.cpython-313.pyc
lib/python3.13/site-packages/flax/__pycache__/io.cpython-313.pyc
lib/python3.13/site-packages/flax/__pycache__/jax_utils.cpython-313.pyc
lib/python3.13/site-packages/flax/__pycache__/serialization.cpython-313.pyc
lib/python3.13/site-packages/flax/__pycache__/struct.cpython-313.pyc
lib/python3.13/site-packages/flax/__pycache__/traceback_util.cpython-313.pyc
lib/python3.13/site-packages/flax/__pycache__/traverse_util.cpython-313.pyc
lib/python3.13/site-packages/flax/__pycache__/typing.cpython-313.pyc
lib/python3.13/site-packages/flax/__pycache__/version.cpython-313.pyc
lib/python3.13/site-packages/flax/configurations.py
lib/python3.13/site-packages/flax/core/__init__.py
lib/python3.13/site-packages/flax/core/__pycache__/__init__.cpython-313.pyc
lib/python3.13/site-packages/flax/core/__pycache__/axes_scan.cpython-313.pyc
lib/python3.13/site-packages/flax/core/__pycache__/frozen_dict.cpython-313.pyc
lib/python3.13/site-packages/flax/core/__pycache__/lift.cpython-313.pyc
lib/python3.13/site-packages/flax/core/__pycache__/meta.cpython-313.pyc
lib/python3.13/site-packages/flax/core/__pycache__/partial_eval.cpython-313.pyc
lib/python3.13/site-packages/flax/core/__pycache__/scope.cpython-313.pyc
lib/python3.13/site-packages/flax/core/__pycache__/spmd.cpython-313.pyc
lib/python3.13/site-packages/flax/core/__pycache__/tracers.cpython-313.pyc
lib/python3.13/site-packages/flax/core/__pycache__/variables.cpython-313.pyc
lib/python3.13/site-packages/flax/core/axes_scan.py
lib/python3.13/site-packages/flax/core/flax_functional_engine.ipynb
lib/python3.13/site-packages/flax/core/frozen_dict.py
lib/python3.13/site-packages/flax/core/lift.py
lib/python3.13/site-packages/flax/core/meta.py
lib/python3.13/site-packages/flax/core/nn/__init__.py
lib/python3.13/site-packages/flax/core/nn/__pycache__/__init__.cpython-313.pyc
lib/python3.13/site-packages/flax/core/nn/__pycache__/attention.cpython-313.pyc
lib/python3.13/site-packages/flax/core/nn/__pycache__/linear.cpython-313.pyc
lib/python3.13/site-packages/flax/core/nn/__pycache__/normalization.cpython-313.pyc
lib/python3.13/site-packages/flax/core/nn/__pycache__/stochastic.cpython-313.pyc
lib/python3.13/site-packages/flax/core/nn/attention.py
lib/python3.13/site-packages/flax/core/nn/linear.py
lib/python3.13/site-packages/flax/core/nn/normalization.py
lib/python3.13/site-packages/flax/core/nn/stochastic.py
lib/python3.13/site-packages/flax/core/partial_eval.py
lib/python3.13/site-packages/flax/core/scope.py
lib/python3.13/site-packages/flax/core/spmd.py
lib/python3.13/site-packages/flax/core/tracers.py
lib/python3.13/site-packages/flax/core/variables.py
lib/python3.13/site-packages/flax/cursor.py
lib/python3.13/site-packages/flax/errors.py
lib/python3.13/site-packages/flax/experimental/__init__.py
lib/python3.13/site-packages/flax/experimental/__pycache__/__init__.cpython-313.pyc
lib/python3.13/site-packages/flax/experimental/__pycache__/nnx.cpython-313.pyc
lib/python3.13/site-packages/flax/experimental/nnx.py
lib/python3.13/site-packages/flax/ids.py
lib/python3.13/site-packages/flax/io.py
lib/python3.13/site-packages/flax/jax_utils.py
lib/python3.13/site-packages/flax/linen/README.md
lib/python3.13/site-packages/flax/linen/__init__.py
lib/python3.13/site-packages/flax/linen/__pycache__/__init__.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/activation.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/attention.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/batch_apply.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/combinators.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/dtypes.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/fp8_ops.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/initializers.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/kw_only_dataclasses.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/linear.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/module.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/normalization.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/partitioning.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/pooling.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/recurrent.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/spmd.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/stochastic.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/summary.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/__pycache__/transforms.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/activation.py
lib/python3.13/site-packages/flax/linen/attention.py
lib/python3.13/site-packages/flax/linen/batch_apply.py
lib/python3.13/site-packages/flax/linen/combinators.py
lib/python3.13/site-packages/flax/linen/dtypes.py
lib/python3.13/site-packages/flax/linen/experimental/__pycache__/layers_with_named_axes.cpython-313.pyc
lib/python3.13/site-packages/flax/linen/experimental/layers_with_named_axes.py
lib/python3.13/site-packages/flax/linen/fp8_ops.py
lib/python3.13/site-packages/flax/linen/initializers.py
lib/python3.13/site-packages/flax/linen/kw_only_dataclasses.py
lib/python3.13/site-packages/flax/linen/linear.py
lib/python3.13/site-packages/flax/linen/module.py
lib/python3.13/site-packages/flax/linen/normalization.py
lib/python3.13/site-packages/flax/linen/partitioning.py
lib/python3.13/site-packages/flax/linen/pooling.py
lib/python3.13/site-packages/flax/linen/recurrent.py
lib/python3.13/site-packages/flax/linen/spmd.py
lib/python3.13/site-packages/flax/linen/stochastic.py
lib/python3.13/site-packages/flax/linen/summary.py
lib/python3.13/site-packages/flax/linen/transforms.py
lib/python3.13/site-packages/flax/metrics/__init__.py
lib/python3.13/site-packages/flax/metrics/__pycache__/__init__.cpython-313.pyc
lib/python3.13/site-packages/flax/metrics/__pycache__/tensorboard.cpython-313.pyc
lib/python3.13/site-packages/flax/metrics/tensorboard.py
lib/python3.13/site-packages/flax/nnx/README.md
lib/python3.13/site-packages/flax/nnx/__init__.py
lib/python3.13/site-packages/flax/nnx/__pycache__/__init__.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/__pycache__/extract.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/__pycache__/filterlib.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/__pycache__/graph.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/__pycache__/helpers.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/__pycache__/ids.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/__pycache__/module.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/__pycache__/proxy_caller.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/__pycache__/pytreelib.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/__pycache__/reprlib.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/__pycache__/rnglib.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/__pycache__/spmd.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/__pycache__/statelib.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/__pycache__/summary.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/__pycache__/tracers.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/__pycache__/traversals.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/__pycache__/variablelib.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/__pycache__/visualization.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/bridge/__init__.py
lib/python3.13/site-packages/flax/nnx/bridge/__pycache__/__init__.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/bridge/__pycache__/interop.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/bridge/__pycache__/module.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/bridge/__pycache__/variables.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/bridge/__pycache__/wrappers.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/bridge/interop.py
lib/python3.13/site-packages/flax/nnx/bridge/module.py
lib/python3.13/site-packages/flax/nnx/bridge/variables.py
lib/python3.13/site-packages/flax/nnx/bridge/wrappers.py
lib/python3.13/site-packages/flax/nnx/extract.py
lib/python3.13/site-packages/flax/nnx/filterlib.py
lib/python3.13/site-packages/flax/nnx/graph.py
lib/python3.13/site-packages/flax/nnx/helpers.py
lib/python3.13/site-packages/flax/nnx/ids.py
lib/python3.13/site-packages/flax/nnx/module.py
lib/python3.13/site-packages/flax/nnx/nn/__init__.py
lib/python3.13/site-packages/flax/nnx/nn/__pycache__/__init__.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/nn/__pycache__/activations.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/nn/__pycache__/attention.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/nn/__pycache__/dtypes.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/nn/__pycache__/initializers.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/nn/__pycache__/linear.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/nn/__pycache__/lora.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/nn/__pycache__/normalization.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/nn/__pycache__/recurrent.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/nn/__pycache__/stochastic.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/nn/activations.py
lib/python3.13/site-packages/flax/nnx/nn/attention.py
lib/python3.13/site-packages/flax/nnx/nn/dtypes.py
lib/python3.13/site-packages/flax/nnx/nn/initializers.py
lib/python3.13/site-packages/flax/nnx/nn/linear.py
lib/python3.13/site-packages/flax/nnx/nn/lora.py
lib/python3.13/site-packages/flax/nnx/nn/normalization.py
lib/python3.13/site-packages/flax/nnx/nn/recurrent.py
lib/python3.13/site-packages/flax/nnx/nn/stochastic.py
lib/python3.13/site-packages/flax/nnx/proxy_caller.py
lib/python3.13/site-packages/flax/nnx/pytreelib.py
lib/python3.13/site-packages/flax/nnx/reprlib.py
lib/python3.13/site-packages/flax/nnx/rnglib.py
lib/python3.13/site-packages/flax/nnx/scripts/requirements.txt
lib/python3.13/site-packages/flax/nnx/scripts/run-all-examples.bash
lib/python3.13/site-packages/flax/nnx/spmd.py
lib/python3.13/site-packages/flax/nnx/statelib.py
lib/python3.13/site-packages/flax/nnx/summary.py
lib/python3.13/site-packages/flax/nnx/tracers.py
lib/python3.13/site-packages/flax/nnx/training/__init__.py
lib/python3.13/site-packages/flax/nnx/training/__pycache__/__init__.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/training/__pycache__/metrics.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/training/__pycache__/optimizer.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/training/metrics.py
lib/python3.13/site-packages/flax/nnx/training/optimizer.py
lib/python3.13/site-packages/flax/nnx/transforms/__init__.py
lib/python3.13/site-packages/flax/nnx/transforms/__pycache__/__init__.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/transforms/__pycache__/autodiff.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/transforms/__pycache__/compilation.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/transforms/__pycache__/general.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/transforms/__pycache__/iteration.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/transforms/__pycache__/transforms.cpython-313.pyc
lib/python3.13/site-packages/flax/nnx/transforms/autodiff.py
lib/python3.13/site-packages/flax/nnx/transforms/compilation.py
lib/python3.13/site-packages/flax/nnx/transforms/general.py
lib/python3.13/site-packages/flax/nnx/transforms/iteration.py
lib/python3.13/site-packages/flax/nnx/transforms/transforms.py
lib/python3.13/site-packages/flax/nnx/traversals.py
lib/python3.13/site-packages/flax/nnx/variablelib.py
lib/python3.13/site-packages/flax/nnx/visualization.py
lib/python3.13/site-packages/flax/oss/ .git-blame-ignore-revs
lib/python3.13/site-packages/flax/py.typed
lib/python3.13/site-packages/flax/serialization.py
lib/python3.13/site-packages/flax/struct.py
lib/python3.13/site-packages/flax/testing/__init__.py
lib/python3.13/site-packages/flax/testing/__pycache__/__init__.cpython-313.pyc
lib/python3.13/site-packages/flax/testing/__pycache__/benchmark.cpython-313.pyc
lib/python3.13/site-packages/flax/testing/benchmark.py
lib/python3.13/site-packages/flax/traceback_util.py
lib/python3.13/site-packages/flax/training/__init__.py
lib/python3.13/site-packages/flax/training/__pycache__/__init__.cpython-313.pyc
lib/python3.13/site-packages/flax/training/__pycache__/checkpoints.cpython-313.pyc
lib/python3.13/site-packages/flax/training/__pycache__/common_utils.cpython-313.pyc
lib/python3.13/site-packages/flax/training/__pycache__/dynamic_scale.cpython-313.pyc
lib/python3.13/site-packages/flax/training/__pycache__/early_stopping.cpython-313.pyc
lib/python3.13/site-packages/flax/training/__pycache__/lr_schedule.cpython-313.pyc
lib/python3.13/site-packages/flax/training/__pycache__/orbax_utils.cpython-313.pyc
lib/python3.13/site-packages/flax/training/__pycache__/prefetch_iterator.cpython-313.pyc
lib/python3.13/site-packages/flax/training/__pycache__/train_state.cpython-313.pyc
lib/python3.13/site-packages/flax/training/checkpoints.py
lib/python3.13/site-packages/flax/training/common_utils.py
lib/python3.13/site-packages/flax/training/dynamic_scale.py
lib/python3.13/site-packages/flax/training/early_stopping.py
lib/python3.13/site-packages/flax/training/lr_schedule.py
lib/python3.13/site-packages/flax/training/orbax_utils.py
lib/python3.13/site-packages/flax/training/prefetch_iterator.py
lib/python3.13/site-packages/flax/training/train_state.py
lib/python3.13/site-packages/flax/traverse_util.py
lib/python3.13/site-packages/flax/typing.py
lib/python3.13/site-packages/flax/version.py
lib/python3.13/site-packages/flaxlib_src/.gitignore
lib/python3.13/site-packages/flaxlib_src/CMakeLists.txt
lib/python3.13/site-packages/flaxlib_src/Cargo.lock
lib/python3.13/site-packages/flaxlib_src/Cargo.toml
lib/python3.13/site-packages/flaxlib_src/LICENSE
lib/python3.13/site-packages/flaxlib_src/README.md
lib/python3.13/site-packages/flaxlib_src/pyproject.toml
lib/python3.13/site-packages/flaxlib_src/src/flaxlib/__init__.py
lib/python3.13/site-packages/flaxlib_src/src/flaxlib/__pycache__/__init__.cpython-313.pyc
lib/python3.13/site-packages/flaxlib_src/src/flaxlib/flaxlib_cpp.pyi
lib/python3.13/site-packages/flaxlib_src/src/lib.cc
lib/python3.13/site-packages/flaxlib_src/uv.lock
