google / flax
Files Coverage
__init__.py 100.00%
config.py 76.92%
core/__init__.py 100.00%
core/axes_scan.py 95.45%
core/frozen_dict.py 96.19%
core/lift.py 95.83%
core/nn/__init__.py 100.00%
core/nn/attention.py 13.47%
core/nn/linear.py 42.34%
core/nn/normalization.py 46.67%
core/nn/stochastic.py 85.71%
core/scope.py 89.98%
core/tracers.py 84.62%
core/variables.py 0.00%
errors.py 90.54%
jax_utils.py 56.38%
linen/__init__.py 100.00%
linen/activation.py 100.00%
linen/attention.py 95.80%
linen/combinators.py 100.00%
linen/dotgetter.py 72.92%
linen/initializers.py 100.00%
linen/linear.py 98.84%
linen/module.py 94.00%
linen/normalization.py 96.03%
linen/partitioning.py 81.50%
linen/pooling.py 84.38%
linen/recurrent.py 100.00%
linen/stochastic.py 96.55%
linen/transforms.py 94.26%
metrics/tensorboard.py 91.30%
optim/__init__.py 100.00%
optim/adabelief.py 50.00%
optim/adadelta.py 56.25%
optim/adafactor.py 31.68%
optim/adagrad.py 63.64%
optim/adam.py 50.00%
optim/base.py 34.10%
optim/dynamic_scale.py 0.00%
optim/lamb.py 43.90%
optim/lars.py 44.74%
optim/momentum.py 50.00%
optim/rmsprop.py 54.84%
optim/sgd.py 56.25%
optim/weight_norm.py 27.85%
serialization.py 67.88%
struct.py 77.05%
testing/__init__.py 0.00%
testing/benchmark.py 0.00%
traceback_util.py 100.00%
training/__init__.py 100.00%
training/checkpoints.py 97.16%
training/common_utils.py 0.00%
training/early_stopping.py 100.00%
training/lr_schedule.py 0.00%
training/prefetch_iterator.py 0.00%
training/train_state.py 0.00%
traverse_util.py 98.97%
version.py 100.00%
Folder Totals (59 files) 74.97%
Project Totals (59 files) 74.97%
Loading