google / flax
Files Coverage
__init__.py 100.00%
core/__init__.py 100.00%
core/axes_scan.py 76.56%
core/frozen_dict.py 98.94%
core/lift.py 96.15%
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 93.73%
core/tracers.py 83.33%
core/variables.py 0.00%
errors.py 88.14%
jax_utils.py 60.58%
linen/__init__.py 100.00%
linen/activation.py 100.00%
linen/attention.py 96.52%
linen/dotgetter.py 70.21%
linen/initializers.py 100.00%
linen/linear.py 99.44%
linen/module.py 94.21%
linen/normalization.py 92.98%
linen/pooling.py 84.38%
linen/recurrent.py 100.00%
linen/stochastic.py 92.31%
linen/transforms.py 91.45%
metrics/tensorboard.py 73.24%
nn/__init__.py 100.00%
nn/activation.py 100.00%
nn/attention.py 75.24%
nn/base.py 92.48%
nn/initializers.py 100.00%
nn/linear.py 99.22%
nn/normalization.py 85.19%
nn/pooling.py 100.00%
nn/recurrent.py 100.00%
nn/stochastic.py 71.05%
nn/utils.py 93.33%
optim/__init__.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 83.98%
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 90.12%
serialization.py 90.06%
struct.py 96.23%
testing/__init__.py 0.00%
testing/benchmark.py 0.00%
training/__init__.py 100.00%
training/checkpoints.py 98.37%
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.68%
version.py 100.00%
Folder Totals (65 files) 82.30%
Project Totals (65 files) 82.30%
Loading