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 95.45%
flax/core/frozen_dict.py 96.19%
flax/core/lift.py 95.92%
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 89.98%
flax/core/tracers.py 84.62%
flax/core/variables.py 0.00%
flax/errors.py 90.54%
flax/jax_utils.py 66.67%
flax/linen/__init__.py 100.00%
flax/linen/activation.py 100.00%
flax/linen/attention.py 95.97%
flax/linen/combinators.py 100.00%
flax/linen/dotgetter.py 72.92%
flax/linen/dtypes.py 94.44%
flax/linen/initializers.py 100.00%
flax/linen/linear.py 97.45%
flax/linen/module.py 94.00%
flax/linen/normalization.py 96.27%
flax/linen/partitioning.py 82.33%
flax/linen/pooling.py 84.38%
flax/linen/recurrent.py 100.00%
flax/linen/stochastic.py 96.55%
flax/linen/summary.py 93.75%
flax/linen/transforms.py 94.44%
flax/metrics/tensorboard.py 91.30%
flax/optim/__init__.py 100.00%
flax/optim/adabelief.py 50.00%
flax/optim/adadelta.py 56.25%
flax/optim/adafactor.py 31.68%
flax/optim/adagrad.py 63.64%
flax/optim/adam.py 50.00%
flax/optim/base.py 34.10%
flax/optim/dynamic_scale.py 0.00%
flax/optim/lamb.py 43.90%
flax/optim/lars.py 44.74%
flax/optim/momentum.py 50.00%
flax/optim/rmsprop.py 54.84%
flax/optim/sgd.py 56.25%
flax/optim/weight_norm.py 27.85%
flax/serialization.py 67.88%
flax/struct.py 77.05%
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 96.50%
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.97%
flax/version.py 100.00%
Project Totals (61 files) 75.90%
Loading