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

flax/core/nn/__init__.py changed.

67.88% ø -22.18%

flax/serialization.py changed.

42.34% ø ø

flax/core/nn/linear.py changed.

Newly tracked file

76.92% ø ø

flax/config.py created.

flax/training/early_stopping.py changed.

flax/__init__.py changed.

85.71% ø ø

flax/core/nn/stochastic.py changed.

flax/linen/activation.py changed.

flax/linen/__init__.py changed.

flax/training/common_utils.py changed.

flax/training/lr_schedule.py changed.

66.67% ø +6.09%

flax/jax_utils.py changed.

56.25% ø -43.75%

flax/optim/sgd.py changed.

flax/version.py changed.

Newly tracked file

82.33% ø ø

flax/linen/partitioning.py created.

flax/core/__init__.py changed.

44.74% ø ø

flax/optim/lars.py changed.

flax/training/__init__.py changed.

Newly tracked file

flax/traceback_util.py created.

46.67% ø ø

flax/core/nn/normalization.py changed.

Newly tracked file

flax/linen/combinators.py created.

96.55% ø +4.24%

flax/linen/stochastic.py changed.

flax/training/train_state.py changed.

77.05% ø -19.18%

flax/struct.py changed.

96.27% ø +3.29%

flax/linen/normalization.py changed.

72.92% ø +2.70%

flax/linen/dotgetter.py changed.

0.00% ø -95.35%

flax/optim/dynamic_scale.py changed.

98.97% ø +0.29%

flax/traverse_util.py changed.

flax/linen/recurrent.py changed.

63.64% ø -36.36%

flax/optim/adagrad.py changed.

34.10% ø -49.88%

flax/optim/base.py changed.

31.68% ø -66.34%

flax/optim/adafactor.py changed.

54.84% ø -45.16%

flax/optim/rmsprop.py changed.

96.19% ø -2.75%

flax/core/frozen_dict.py changed.

95.45% ø +18.89%

flax/core/axes_scan.py changed.

43.90% ø ø

flax/optim/lamb.py changed.

flax/optim/__init__.py changed.

flax/testing/benchmark.py changed.

27.85% ø -62.28%

flax/optim/weight_norm.py changed.

97.45% ø -1.98%

flax/linen/linear.py changed.

89.98% ø -3.76%

flax/core/scope.py changed.

Newly tracked file

50.00% ø ø

flax/optim/adabelief.py created.

50.00% ø -50.00%

flax/optim/adam.py changed.

13.47% ø ø

flax/core/nn/attention.py changed.

50.00% ø -43.33%

flax/optim/momentum.py changed.

flax/testing/__init__.py changed.

97.16% ø -1.21%

flax/training/checkpoints.py changed.

90.54% ø +2.40%

flax/errors.py changed.

flax/linen/initializers.py changed.

95.92% ø -0.24%

flax/core/lift.py changed.

flax/core/variables.py changed.

56.25% ø -43.75%

flax/optim/adadelta.py changed.

95.97% ø -0.55%

flax/linen/attention.py changed.

flax/training/prefetch_iterator.py changed.

Other files ignored by Codecov

examples/ppo/agent.py has changed.

docs/mission.md is new.

examples/nlp_seq/input_pipeline.py has changed.

flax/optim/README.md has changed.

examples/wmt/configs/gpu_mixed_precision.py is new.

examples/sst2/train.py has changed.

examples/sst2/main.py has changed.

examples/ogbg_molpcba/main.py has changed.

examples/linen_design_test/weight_std.py has changed.

examples/pixelcnn/train.py was deleted.

examples/lm1b/requirements.txt has changed.

examples/linen_design_test/mlp_explicit.py has changed.

examples/cloud/README.md has changed.

examples/ppo/ppo_lib.py has changed.

examples/pixelcnn/pixelcnn.py was deleted.

examples/imagenet/input_pipeline.py has changed.

examples/vae/train.py has changed.

tests/core/core_scope_test.py has changed.

examples/ppo/README.md has changed.

docs/howtos/optax_update_guide.rst is new.

examples/imagenet/imagenet_benchmark.py has changed.

flax/linen/experimental/layers_with_named_axes.py is new.

flax/nn/initializers.py was deleted.

flax/nn/__init__.py was deleted.

tests/core/core_lift_test.py is new.

docs/_templates/autosummary/flax_module.rst is new.

images/flax_logo.svg is new.

docs/notebooks/full_eval.md is new.

examples/sst2/README.md has changed.

setup.py has changed.

examples/linen_design_test/autoencoder.py has changed.

examples/lm1b/main.py has changed.

examples/lm1b/models.py has changed.

tests/core/lift_test.py was deleted.

docs/overview.md has changed.

examples/ogbg_molpcba/README.md is new.

docs/howtos/model_surgery.rst has changed.

examples/lm1b/input_pipeline.py has changed.

examples/ogbg_molpcba/models_test.py is new.

CHANGELOG.md has changed.

.pre-commit-config.yaml is new.

examples/wmt/tokenizer.py has changed.

examples/nlp_seq/train.py has changed.

examples/imagenet/imagenet_fake_data_benchmark.py has changed.

docs/flax.traverse_util.rst has changed.

images/flax_logo.png is new.

examples/lm1b/train_test.py has changed.

examples/ogbg_molpcba/configs/test.py is new.

docs/notebooks/state_params.ipynb is new.

docs/design_notes/setup_or_nncompact.rst is new.

docs/design_notes/module_lifecycle.rst is new.

tests/core/core_frozen_dict_test.py has changed.

pytest.ini has changed.

docs/notebooks/linen_intro.md is new.

docs/requirements.txt has changed.

flax/nn/utils.py was deleted.

examples/ogbg_molpcba/configs/default.py is new.

examples/vae/results/.gitignore has changed.

examples/sst2/models.py has changed.

images/flax_logo_250px.png is new.

examples/seq2seq/input_pipeline.py is new.

docs/design_notes/arguments.md has changed.

examples/imagenet/configs/default.py has changed.

pytype.cfg is new.

.github/pull_request_template.md has changed.

...amples/imagenet/configs/v100_x8_mixed_precision.py has changed.

flax/metrics/__init__.py has changed.

docs/flax.linen.rst has changed.

docs/flax.config.rst is new.

examples/linen_design_test/linear_regression.py has changed.

docs/notebooks/optax_update_guide.md is new.

examples/seq2seq/requirements.txt has changed.

docs/flax.traceback_util.rst is new.

flax/nn/pooling.py was deleted.

docs/flip/1777-default-dtype.md is new.

flax/nn/base.py was deleted.

examples/ogbg_molpcba/requirements.txt is new.

examples/mnist/configs/default.py has changed.

.github/ISSUE_TEMPLATE/bug_report.md has changed.

examples/pixelcnn/train_test.py was deleted.

examples/ogbg_molpcba/ogbg_molpcba_benchmark.py is new.

examples/ppo/ppo_lib_test.py has changed.

tests/core/design/core_tied_autoencoder_test.py has changed.

docs/flax.training.rst has changed.

examples/wmt/decode.py has changed.

examples/wmt/train_test.py has changed.

examples/wmt/configs/default.py has changed.

examples/ppo/models.py has changed.

docs/deprecated_howtos/index.rst was deleted.

docs/notebooks/annotated_mnist.md is new.

examples/ppo/requirements.txt has changed.

docs/howtos/ensembling.rst has changed.

docs/conf.py has changed.

examples/mnist/mnist.ipynb has changed.

examples/mnist/mnist_benchmark.py has changed.

examples/imagenet/train_test.py has changed.

docs/notebooks/flax_basics.md is new.

examples/nlp_seq/requirements.txt is new.

examples/ogbg_molpcba/configs/hparam_sweep.py is new.

examples/vae/utils.py has changed.

examples/nlp_seq/README.md has changed.

docs/howtos/convert_pytorch_to_flax.rst is new.

docs/installation.md has changed.

examples/ppo/env_utils.py has changed.

docs/_ext/codediff_test.py has changed.

examples/linen_design_test/mlp_lazy.py has changed.

flax/linen/README.md has changed.

tests/checkpoints_test.py has changed.

docs/conf_sphinx_patch.py is new.

examples/mnist/requirements.txt has changed.

docs/deprecated_howtos/checkpointing.rst was deleted.

examples/ogbg_molpcba/ogbg_molpcba.ipynb is new.

docs/notebooks/jax_for_the_impatient.ipynb has changed.

examples/pixelcnn/sample.png was deleted.

docs/howtos/lr_schedule.rst has changed.

flax/nn/normalization.py was deleted.

examples/pixelcnn/input_pipeline.py was deleted.

examples/ogbg_molpcba/input_pipeline_test.py is new.

tests/core/design/core_dense_test.py has changed.

flax/nn/linear.py was deleted.

docs/deprecated_howtos/polyak-averaging.rst was deleted.

flax/nn/activation.py was deleted.

tests/core/design/core_auto_encoder_test.py has changed.

examples/sst2/models_test.py has changed.

examples/vae/requirements.txt has changed.

examples/imagenet/configs/tpu.py has changed.

flax/py.typed is new.

examples/lm1b/train.py has changed.

docs/deprecated_howtos/ensembling.rst was deleted.

examples/imagenet/models.py has changed.

docs/contributing.md has changed.

docs/flax.nn.rst was deleted.

examples/wmt/input_pipeline_test.py has changed.

docs/howtos/extracting_intermediates.rst has changed.

examples/imagenet/configs/v100_x8.py has changed.

docs/README.md has changed.

.github/analytics/pr_data_query.gql is new.

examples/pixelcnn/requirements.txt was deleted.

docs/notebooks/full_eval.ipynb is new.

examples/nlp_seq/models.py has changed.

tests/core/design/core_big_resnets_test.py has changed.

examples/imagenet/models_test.py has changed.

docs/deprecated_howtos/scheduled-sampling.rst was deleted.

images/flax_logo_500px.png is new.

examples/wmt/requirements.txt has changed.

docs/flax.optim.rst has changed.

docs/howtos/full_eval.rst is new.

examples/imagenet/README.md has changed.

.github/workflows/build.yml has changed.

docs/flax.core.frozen_dict.rst has changed.

flax/nn/recurrent.py was deleted.

examples/sst2/input_pipeline.py has changed.

examples/sst2/vocabulary.py has changed.

examples/sst2/configs/default.py has changed.

docs/notebooks/flax_basics.ipynb has changed.

.github/analytics/issue_activity_since_date.gql is new.

README.md has changed.

examples/linen_design_test/mlp_inline.py has changed.

examples/mnist/main.py has changed.

tests/core/design/core_flow_test.py has changed.

examples/ppo/seed_rl_atari_preprocessing.py has changed.

examples/imagenet/main.py has changed.

examples/ppo/configs/default.py has changed.

examples/imagenet/train.py has changed.

tests/core/design/core_resnet_test.py has changed.

docs/flax.jax_utils.rst has changed.

examples/wmt/main.py has changed.

flax/core/flax_functional_engine.ipynb has changed.

flax/nn/attention.py was deleted.

examples/wmt/input_pipeline.py has changed.

flax/nn/stochastic.py was deleted.

tests/core/design/core_vmap_test.py has changed.

examples/ogbg_molpcba/train.py is new.

examples/ogbg_molpcba/input_pipeline.py is new.

examples/imagenet/configs/fake_data_benchmark.py has changed.

docs/deprecated_howtos/howtos-howto.md was deleted.

docs/notebooks/jax_for_the_impatient.md is new.

examples/lm1b/README.md has changed.

examples/ppo/test_episodes.py has changed.

docs/notebooks/state_params.md is new.

examples/linen_design_test/attention_simple.py has changed.

examples/linen_design_test/tied_autoencoder.py has changed.

examples/mnist/train_test.py has changed.

examples/seq2seq/train.py has changed.

tests/core/design/core_weight_std_test.py has changed.

examples/seq2seq/seq2seq.ipynb has changed.

examples/wmt/train.py has changed.

pylintrc has changed.

examples/wmt/bleu.py has changed.

tests/core/design/core_custom_vjp_test.py has changed.

docs/notebooks/linen_intro.ipynb has changed.

docs/flip/1009-optimizer-api.md has changed.

examples/linen_design_test/dense.py has changed.

examples/ppo/ppo_main.py has changed.

examples/sst2/sst2.ipynb has changed.

examples/pixelcnn/configs/default.py was deleted.

examples/lm1b/input_pipeline_test.py has changed.

examples/sst2/train_test.py has changed.

tests/core/design/core_scan_test.py has changed.

examples/pixelcnn/README.md was deleted.

.github/ISSUE_TEMPLATE/example_request.md was deleted.

docs/examples.rst was deleted.

examples/seq2seq/README.md has changed.

docs/notebooks/optax_update_guide.ipynb is new.

examples/pixelcnn/sample.py was deleted.

docs/howtos/linen_upgrade_guide.rst is new.

examples/ogbg_molpcba/train_test.py is new.

examples/README.md has changed.

examples/mnist/train.py has changed.

contributing.md has changed.

docs/deprecated_howtos/distributed-training.rst was deleted.

docs/notebooks/annotated_mnist.ipynb has changed.

examples/cloud/startup_script.sh has changed.

examples/wmt/models.py has changed.

examples/seq2seq/train_test.py has changed.

examples/lm1b/tokenizer.py has changed.

examples/cloud/launch_gce.py has changed.

tests/core/design/core_attention_test.py has changed.

.github/analytics/get_repo_metrics.py is new.

examples/pixelcnn/model_test.py was deleted.

docs/howtos/state_params.rst has changed.

examples/imagenet/imagenet.ipynb has changed.

examples/imagenet/requirements.txt has changed.

examples/nlp_seq/input_pipeline_test.py has changed.

examples/seq2seq/models.py is new.

examples/sst2/requirements.txt has changed.

examples/lm1b/configs/default.py has changed.

examples/ogbg_molpcba/configs/default_graph_net.py is new.

examples/lm1b/temperature_sampler.py has changed.

dev/update_requirements.py is new.

linen_examples/README.md was deleted.

examples/ogbg_molpcba/models.py is new.

docs/design_notes/lift.md has changed.

docs/_ext/codediff.py has changed.

examples/mnist/README.md has changed.

examples/sst2/input_pipeline_test.py has changed.

examples/sst2/build_vocabulary.py has changed.

examples/wmt/README.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

→

Loading file...

Changes in flax/optim/momentum.py

-13

→

+13

Loading file...

Changes in flax/optim/adagrad.py

-7

→

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

→

Loading file...

Changes in flax/linen/linear.py

→

+91

Loading file...

Changes in flax/training/checkpoints.py

→

+16

Loading file...

Changes in flax/linen/attention.py

→

Loading file...

Changes in flax/core/lift.py

→

+73

Loading file...

Changes in flax/linen/module.py

→

+63

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

→

Loading file...

Changes in flax/optim/__init__.py

→

Loading file...

Changes in flax/version.py

→

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

→

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

→

Loading file...

Changes in flax/errors.py

→

+15

Loading file...

Changes in flax/linen/dotgetter.py

-1

→

Loading file...

Changes in flax/linen/transforms.py

→

+131

Loading file...

Changes in flax/linen/normalization.py

-3

→

+23

Loading file...

Changes in flax/linen/stochastic.py

-1

→

Loading file...

Changes in flax/jax_utils.py

→

+21

Loading file...

Changes in flax/metrics/tensorboard.py

-13

→

+11

Loading file...

Changes in flax/core/axes_scan.py

-12

→

+14

Loading file...

Merge pull request #2231 from cgarciae:fix-notebooks

5b4ef40

fix jax_for_the_impatient

#2231 ff01ad5

75.91% ø +0.07%

#2231 6a2e9e3

7f51005

+15

d3a3b36

#2228 c8781de

Merge pull request #2166 from cgarciae:test-notebooks

f647b21

Merge branch 'main' into test-notebooks

#2166 84cd54c

#2166 4572ec0

-15

#2205 7de387d

#2205 bde4caf

+15

017d485

Hiding 5 contexual commits

#2209 d3da1ea

Hiding 1 contexual commits

#2206 025e597

#2206 d22e4f7

#2206 5c2517a

#2203 fb420e2

Add newline at end of .pre-commit-config.yaml

#2166 2761f64

66b4a0e

#1916 3a3df8d

#1916 89ef718

#2166 a485202

myst_nb + jupytext workflow

#2166 c5848a5

Merge pull request #2167 from cgarciae:improve-module-sphinx

75.84% ø -0.01%

43c8057

0d12ea2

-2

#2167 04074a9

898b159

Set correct PyYAML version

#2193 992019c

#2193 506afa4

#2193 2e6caf3

Hiding 1 contexual commits

ef7c5b8

Merge pull request #2188 from PWhiddy:patch-1

a58ee9a

Hiding 1 contexual commits

Merge pull request #2181 from rsokl:patch-1

a557b13

Update docs/notebooks/flax_basics.ipynb

#2181 8cb503f

Update docs/notebooks/flax_basics.ipynb

#2181 1dfbda0

Merge pull request #2180 from cgarciae:0.5.1

500d349

75.85% 100.00% ø

c182c7d

75.85% 100.00% +0.04%

Update normalization.py

yechengxi

#2182 68c47cc

Add missing `@jax.jit` to `mse`

75.85% ø +0.04%

rsokl

#2181 3f07d18

75.85% 100.00% +0.04%

update version to 0.5.1

#2180 aca5b53

Allow nn.cond, nn.while to act on bound methods.

19fc095

Hiding 2 contexual commits

-1

75.81% 100.00% +0.01%

levskaya

#2172 294bb41

Add a reduction_axis argument to LayerNorm. This allows audio models to normalize across a time axis.

#2171 67c09a1

75.80% 100.00% +0.01%

0a5a187

-2

-3

75.83% 100.00% +0.04%

improve tabulate

#2162 f6f906b

-1

f095d13

Updates dynamic_scale in training state.

andsteing

9d4505b

Fixes small typos in design_notes/module_lifecycle.rst

andsteing

b3236ce

#2163 ae7b82d

Merge pull request #2158 from jheek:forward-unroll-arg

8a148db

#2158 49ad422

Merge pull request #2155 from cgarciae:improve-contributing

4c39ba1

Merge pull request #2154 from cgarciae:more-metrics

75.80% ø +0.01%

97e01d7

add docs requirements to contributing guide

Hiding 1 contexual commits

75.80% ø +0.01%

#2155 58f45c8

3bb0433

75.78% ø ø

Merge pull request #2140 from saiteja13427:main

#2154 f8683cf

75.78% ø ø

c22f7d4

Copybara import of the project:

Hiding 1 contexual commits

75.78% ø ø

77c5c47