google / flax
Files Coverage
flax/__init__.py 100.00%
flax/config.py 76.92%
flax/core/__init__.py 100.00%
flax/core/axes_scan.py 90.77%
flax/core/frozen_dict.py 96.19%
flax/core/lift.py 94.97%
flax/core/nn/__init__.py 100.00%
flax/core/nn/attention.py 13.47%
flax/core/nn/linear.py 42.34%
flax/core/nn/normalization.py 46.67%
flax/core/nn/stochastic.py 85.71%
flax/core/scope.py 88.97%
flax/core/tracers.py 84.62%
flax/core/variables.py 0.00%
flax/deprecated/nn/__init__.py 100.00%
flax/deprecated/nn/activation.py 100.00%
flax/deprecated/nn/attention.py 75.24%
flax/deprecated/nn/base.py 92.48%
flax/deprecated/nn/initializers.py 100.00%
flax/deprecated/nn/linear.py 99.22%
flax/deprecated/nn/normalization.py 85.19%
flax/deprecated/nn/pooling.py 100.00%
flax/deprecated/nn/recurrent.py 100.00%
flax/deprecated/nn/stochastic.py 71.05%
flax/deprecated/nn/utils.py 93.33%
flax/errors.py 88.24%
flax/jax_utils.py 70.97%
flax/linen/__init__.py 100.00%
flax/linen/activation.py 100.00%
flax/linen/attention.py 95.69%
flax/linen/dotgetter.py 70.21%
flax/linen/initializers.py 100.00%
flax/linen/linear.py 99.05%
flax/linen/module.py 93.72%
flax/linen/normalization.py 96.69%
flax/linen/partitioning.py 78.79%
flax/linen/pooling.py 84.38%
flax/linen/recurrent.py 100.00%
flax/linen/stochastic.py 92.00%
flax/linen/transforms.py 94.15%
flax/metrics/tensorboard.py 73.24%
flax/optim/__init__.py 100.00%
flax/optim/adabelief.py 100.00%
flax/optim/adadelta.py 100.00%
flax/optim/adafactor.py 98.02%
flax/optim/adagrad.py 100.00%
flax/optim/adam.py 100.00%
flax/optim/base.py 80.81%
flax/optim/dynamic_scale.py 95.35%
flax/optim/lamb.py 43.90%
flax/optim/lars.py 44.74%
flax/optim/momentum.py 93.33%
flax/optim/rmsprop.py 100.00%
flax/optim/sgd.py 100.00%
flax/optim/weight_norm.py 91.14%
flax/serialization.py 93.21%
flax/struct.py 92.06%
flax/testing/__init__.py 0.00%
flax/testing/benchmark.py 0.00%
flax/traceback_util.py 100.00%
flax/training/__init__.py 100.00%
flax/training/checkpoints.py 97.89%
flax/training/common_utils.py 0.00%
flax/training/early_stopping.py 100.00%
flax/training/lr_schedule.py 0.00%
flax/training/prefetch_iterator.py 0.00%
flax/training/train_state.py 0.00%
flax/traverse_util.py 98.93%
flax/version.py 100.00%
Project Totals (69 files) 83.17%
Loading