google / flax

Compare 1844de4 ... +249 ... 5b4ef40

Showing 61 of 300 files from the diff.
Newly tracked file
flax/linen/dtypes.py created.
Newly tracked file
flax/linen/summary.py created.
Newly tracked file
flax/config.py created.
Newly tracked file
flax/traceback_util.py created.
Newly tracked file
flax/optim/adabelief.py created.
Other files ignored by Codecov
setup.py has changed.
docs/overview.md has changed.
CHANGELOG.md has changed.
pytest.ini has changed.
flax/nn/utils.py was deleted.
pytype.cfg is new.
flax/nn/base.py was deleted.
docs/conf.py has changed.
flax/nn/linear.py was deleted.
docs/flax.nn.rst was deleted.
docs/README.md has changed.
README.md has changed.
pylintrc has changed.
docs/examples.rst was deleted.
contributing.md has changed.
docs/index.rst has changed.

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Click to load this diff.
Loading diff...

Learn more Showing 51 files with coverage changes found.

Changes in flax/optim/dynamic_scale.py
-41
+41
Loading file...
Changes in flax/optim/adafactor.py
-67
+67
Loading file...
Changes in flax/optim/weight_norm.py
-51
+49
Loading file...
Changes in flax/optim/adam.py
-18
+18
Loading file...
Changes in flax/optim/base.py
-114
+81
Loading file...
Changes in flax/optim/rmsprop.py
-13
+14
Loading file...
Changes in flax/optim/adadelta.py
-13
+14
Loading file...
Changes in flax/optim/sgd.py
-7
+7
Loading file...
Changes in flax/optim/momentum.py
-13
+13
Loading file...
Changes in flax/optim/adagrad.py
-7
+8
Loading file...
Changes in flax/serialization.py
-33
+37
Loading file...
Changes in flax/struct.py
-4
+12
Loading file...
Changes in flax/core/scope.py
+63
+21
Loading file...
Changes in flax/core/frozen_dict.py
+8
+3
Loading file...
Changes in flax/linen/linear.py
+91
+6
Loading file...
Changes in flax/training/checkpoints.py
+16
+2
Loading file...
Changes in flax/linen/attention.py
+8
+1
Loading file...
Changes in flax/core/lift.py
+73
+4
Loading file...
Changes in flax/linen/module.py
+63
+5
Loading file...
Changes in flax/linen/recurrent.py
+19
Loading file...
Changes in flax/__init__.py
-1
Loading file...
Changes in flax/core/nn/attention.py
New
Loading file...
Changes in flax/linen/activation.py
+16
Loading file...
Changes in flax/linen/__init__.py
+2
Loading file...
Changes in flax/optim/__init__.py
+2
Loading file...
Changes in flax/version.py
+1
Loading file...
Changes in flax/training/__init__.py
New
Loading file...
Changes in flax/core/__init__.py
New
Loading file...
Changes in flax/core/nn/__init__.py
New
Loading file...
Changes in flax/linen/initializers.py
+5
Loading file...
Changes in flax/testing/__init__.py
New
Loading file...
Changes in flax/training/early_stopping.py
New
Loading file...
Changes in flax/testing/benchmark.py
New
Loading file...
Changes in flax/core/nn/linear.py
New
Loading file...
New file flax/linen/summary.py
New
Loading file...
New file flax/traceback_util.py
New
Loading file...
New file flax/config.py
New
Loading file...
New file flax/optim/adabelief.py
New
Loading file...
New file flax/linen/dtypes.py
New
Loading file...
New file flax/linen/combinators.py
New
Loading file...
New file flax/linen/partitioning.py
New
Loading file...
Changes in flax/traverse_util.py
+43
Loading file...
Changes in flax/core/tracers.py
+1
Loading file...
Changes in flax/errors.py
+15
Loading file...
Changes in flax/linen/dotgetter.py
-1
+2
Loading file...
Changes in flax/linen/transforms.py
+131
+4
Loading file...
Changes in flax/linen/normalization.py
-3
+23
Loading file...
Changes in flax/linen/stochastic.py
-1
+4
Loading file...
Changes in flax/jax_utils.py
+21
+1
Loading file...
Changes in flax/metrics/tensorboard.py
-13
+11
Loading file...
Changes in flax/core/axes_scan.py
-12
+14
Loading file...

251 Commits

+15
+15
-15
-15
+15
+15
Hiding 5 contexual commits Hiding 1 contexual commits
-2
-2
Hiding 1 contexual commits
Hiding 1 contexual commits
Hiding 2 contexual commits
+4
+5
-1
+2
+2
-2
-3
+1
+4
+5
-1
Hiding 1 contexual commits
+7
+6
+1
Hiding 1 contexual commits