site-packages/flax-0.5.1.dist-info/AUTHORS
site-packages/flax-0.5.1.dist-info/INSTALLER
site-packages/flax-0.5.1.dist-info/LICENSE
site-packages/flax-0.5.1.dist-info/METADATA
site-packages/flax-0.5.1.dist-info/RECORD
site-packages/flax-0.5.1.dist-info/REQUESTED
site-packages/flax-0.5.1.dist-info/WHEEL
site-packages/flax-0.5.1.dist-info/direct_url.json
site-packages/flax-0.5.1.dist-info/top_level.txt
site-packages/flax/__init__.py
site-packages/flax/config.py
site-packages/flax/core/__init__.py
site-packages/flax/core/axes_scan.py
site-packages/flax/core/frozen_dict.py
site-packages/flax/core/lift.py
site-packages/flax/core/nn/__init__.py
site-packages/flax/core/nn/attention.py
site-packages/flax/core/nn/linear.py
site-packages/flax/core/nn/normalization.py
site-packages/flax/core/nn/stochastic.py
site-packages/flax/core/scope.py
site-packages/flax/core/tracers.py
site-packages/flax/core/variables.py
site-packages/flax/errors.py
site-packages/flax/jax_utils.py
site-packages/flax/linen/__init__.py
site-packages/flax/linen/activation.py
site-packages/flax/linen/attention.py
site-packages/flax/linen/combinators.py
site-packages/flax/linen/dotgetter.py
site-packages/flax/linen/dtypes.py
site-packages/flax/linen/initializers.py
site-packages/flax/linen/linear.py
site-packages/flax/linen/module.py
site-packages/flax/linen/normalization.py
site-packages/flax/linen/partitioning.py
site-packages/flax/linen/pooling.py
site-packages/flax/linen/recurrent.py
site-packages/flax/linen/stochastic.py
site-packages/flax/linen/summary.py
site-packages/flax/linen/transforms.py
site-packages/flax/metrics/__init__.py
site-packages/flax/metrics/tensorboard.py
site-packages/flax/optim/__init__.py
site-packages/flax/optim/adabelief.py
site-packages/flax/optim/adadelta.py
site-packages/flax/optim/adafactor.py
site-packages/flax/optim/adagrad.py
site-packages/flax/optim/adam.py
site-packages/flax/optim/base.py
site-packages/flax/optim/dynamic_scale.py
site-packages/flax/optim/lamb.py
site-packages/flax/optim/lars.py
site-packages/flax/optim/momentum.py
site-packages/flax/optim/rmsprop.py
site-packages/flax/optim/sgd.py
site-packages/flax/optim/weight_norm.py
site-packages/flax/py.typed
site-packages/flax/serialization.py
site-packages/flax/struct.py
site-packages/flax/testing/__init__.py
site-packages/flax/testing/benchmark.py
site-packages/flax/traceback_util.py
site-packages/flax/training/__init__.py
site-packages/flax/training/checkpoints.py
site-packages/flax/training/common_utils.py
site-packages/flax/training/early_stopping.py
site-packages/flax/training/lr_schedule.py
site-packages/flax/training/prefetch_iterator.py
site-packages/flax/training/train_state.py
site-packages/flax/traverse_util.py
site-packages/flax/version.py
