google / flax
Files Coverage
__init__.py 100.00%
config.py 76.92%
core/__init__.py 100.00%
core/axes_scan.py 90.77%
core/frozen_dict.py 96.19%
core/lift.py 94.97%
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.10%
core/tracers.py 84.62%
core/variables.py 0.00%
deprecated/nn/__init__.py 100.00%
deprecated/nn/activation.py 100.00%
deprecated/nn/attention.py 75.24%
deprecated/nn/base.py 92.48%
deprecated/nn/initializers.py 100.00%
deprecated/nn/linear.py 99.22%
deprecated/nn/normalization.py 85.19%
deprecated/nn/pooling.py 100.00%
deprecated/nn/recurrent.py 100.00%
deprecated/nn/stochastic.py 71.05%
deprecated/nn/utils.py 93.33%
errors.py 88.24%
jax_utils.py 70.97%
linen/__init__.py 100.00%
linen/activation.py 100.00%
linen/attention.py 95.69%
linen/dotgetter.py 70.21%
linen/initializers.py 100.00%
linen/linear.py 99.05%
linen/module.py 93.72%
linen/normalization.py 96.69%
linen/partitioning.py 78.79%
linen/pooling.py 84.38%
linen/recurrent.py 100.00%
linen/stochastic.py 92.00%
linen/transforms.py 94.15%
metrics/tensorboard.py 73.24%
optim/__init__.py 100.00%
optim/adabelief.py 100.00%
optim/adadelta.py 100.00%
optim/adafactor.py 98.02%
optim/adagrad.py 100.00%
optim/adam.py 100.00%
optim/base.py 80.81%
optim/dynamic_scale.py 95.35%
optim/lamb.py 43.90%
optim/lars.py 44.74%
optim/momentum.py 93.33%
optim/rmsprop.py 100.00%
optim/sgd.py 100.00%
optim/weight_norm.py 91.14%
serialization.py 93.21%
struct.py 92.06%
testing/__init__.py 0.00%
testing/benchmark.py 0.00%
traceback_util.py 100.00%
training/__init__.py 100.00%
training/checkpoints.py 97.89%
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.93%
version.py 100.00%
Folder Totals (69 files) 83.18%
Project Totals (69 files) 83.18%
Loading