google / flax

Compare a591f96 ... +49 ... 64471a0

83.17% 90.35% +0.05%

Overview Diff Coverage Changes Files Commits
Coverage Reach
linen/module.py linen/linear.py linen/transforms.py linen/partitioning.py linen/recurrent.py linen/normalization.py linen/attention.py linen/dotgetter.py linen/pooling.py linen/activation.py linen/stochastic.py linen/initializers.py linen/__init__.py core/scope.py core/nn/attention.py core/nn/linear.py core/nn/normalization.py core/nn/stochastic.py core/nn/__init__.py core/lift.py core/frozen_dict.py core/axes_scan.py core/tracers.py core/__init__.py core/variables.py deprecated/nn/base.py deprecated/nn/attention.py deprecated/nn/linear.py deprecated/nn/recurrent.py deprecated/nn/normalization.py deprecated/nn/utils.py deprecated/nn/stochastic.py deprecated/nn/activation.py deprecated/nn/pooling.py deprecated/nn/initializers.py deprecated/nn/__init__.py optim/base.py optim/adafactor.py optim/weight_norm.py optim/dynamic_scale.py optim/lamb.py optim/lars.py optim/adam.py optim/adabelief.py optim/adadelta.py optim/momentum.py optim/rmsprop.py optim/adagrad.py optim/sgd.py optim/__init__.py training/checkpoints.py training/prefetch_iterator.py training/lr_schedule.py training/common_utils.py training/train_state.py training/early_stopping.py training/__init__.py traverse_util.py serialization.py testing/benchmark.py testing/__init__.py jax_utils.py metrics/tensorboard.py errors.py struct.py traceback_util.py config.py __init__.py version.py

No flags found

Use flags to group coverage reports by test type, project and/or folders.
Then setup custom commit statuses and notifications for each flag.

e.g., #unittest #integration

#production #enterprise

#frontend #backend

Learn more about Codecov Flags here.

Showing 68 of 220 files from the diff.
100.00% ø ø
flax/optim/adabelief.py changed.
84.62% ø ø
flax/core/tracers.py changed.
100.00% ø ø
flax/traceback_util.py changed.
75.24% ø ø
flax/deprecated/nn/attention.py changed.
100.00% ø ø
flax/__init__.py changed.
73.24% ø ø
flax/metrics/tensorboard.py changed.
100.00% ø ø
flax/deprecated/nn/__init__.py changed.
80.81% ø ø
flax/optim/base.py changed.
93.72% 100.00% +0.21%
flax/linen/module.py changed.
94.15% 100.00% +0.30%
flax/linen/transforms.py changed.
100.00% ø ø
flax/deprecated/nn/activation.py changed.
93.33% ø ø
flax/deprecated/nn/utils.py changed.
93.21% ø ø
flax/serialization.py changed.
42.34% ø ø
flax/core/nn/linear.py changed.
91.14% ø ø
flax/optim/weight_norm.py changed.
76.92% ø ø
flax/config.py changed.
100.00% ø ø
flax/training/early_stopping.py changed.
98.02% ø ø
flax/optim/adafactor.py changed.
100.00% ø ø
flax/linen/activation.py changed.
100.00% 100.00% ø
flax/linen/__init__.py changed.
0.00% ø ø
flax/training/lr_schedule.py changed.
70.97% ø ø
flax/jax_utils.py changed.
100.00% ø ø
flax/optim/sgd.py changed.
100.00% ø ø
flax/deprecated/nn/pooling.py changed.
78.79% ø ø
flax/linen/partitioning.py changed.
100.00% ø ø
flax/core/__init__.py changed.
84.38% ø ø
flax/linen/pooling.py changed.
100.00% ø ø
flax/linen/initializers.py changed.
46.67% ø ø
flax/core/nn/normalization.py changed.
92.00% ø ø
flax/linen/stochastic.py changed.
99.22% ø ø
flax/deprecated/nn/linear.py changed.
0.00% ø ø
flax/testing/benchmark.py changed.
0.00% ø ø
flax/training/train_state.py changed.
94.97% 100.00% +0.01%
flax/core/lift.py changed.
96.69% 100.00% +0.08%
flax/linen/normalization.py changed.
70.21% ø ø
flax/linen/dotgetter.py changed.
95.35% 100.00% ø
flax/optim/dynamic_scale.py changed.
98.93% ø ø
flax/traverse_util.py changed.
100.00% 100.00% ø
flax/linen/recurrent.py changed.
97.89% ø ø
flax/training/checkpoints.py changed.
100.00% ø ø
flax/optim/adagrad.py changed.
100.00% ø ø
flax/optim/__init__.py changed.
100.00% ø ø
flax/optim/rmsprop.py changed.
96.19% ø ø
flax/core/frozen_dict.py changed.
90.77% ø ø
flax/core/axes_scan.py changed.
100.00% ø ø
flax/training/__init__.py changed.
0.00% ø ø
flax/training/common_utils.py changed.
99.05% 100.00% +0.04%
flax/linen/linear.py changed.
88.97% 56.00% -1.52%
flax/core/scope.py changed.
85.19% ø ø
flax/deprecated/nn/normalization.py changed.
100.00% ø ø
flax/optim/adam.py changed.
71.05% ø ø
flax/deprecated/nn/stochastic.py changed.
93.33% ø ø
flax/optim/momentum.py changed.
0.00% ø ø
flax/testing/__init__.py changed.
13.47% ø ø
flax/core/nn/attention.py changed.
100.00% ø ø
flax/optim/adadelta.py changed.
92.48% ø ø
flax/deprecated/nn/base.py changed.
43.90% ø ø
flax/optim/lamb.py changed.
100.00% ø ø
flax/core/nn/__init__.py changed.
95.69% 100.00% +0.04%
flax/linen/attention.py changed.
44.74% ø ø
flax/optim/lars.py changed.
100.00% ø ø
flax/deprecated/nn/initializers.py changed.
0.00% ø ø
flax/core/variables.py changed.
92.06% ø ø
flax/struct.py changed.
85.71% ø ø
flax/core/nn/stochastic.py changed.
88.24% ø ø
flax/errors.py changed.
0.00% ø ø
flax/training/prefetch_iterator.py changed.
100.00% ø ø
flax/deprecated/nn/recurrent.py changed.
Other files ignored by Codecov
examples/ppo/agent.py has changed.
examples/nlp_seq/input_pipeline.py has changed.
examples/wmt/configs/gpu_mixed_precision.py has changed.
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 has changed.
examples/lm1b/requirements.txt has changed.
examples/linen_design_test/mlp_explicit.py has changed.
examples/ppo/ppo_lib.py has changed.
examples/pixelcnn/pixelcnn.py has changed.
examples/imagenet/input_pipeline.py has changed.
examples/vae/train.py has changed.
tests/core/core_scope_test.py has changed.
tests/linen/linen_activation_test.py has changed.
examples/imagenet/imagenet_benchmark.py has changed.
tests/core/core_lift_test.py 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.
examples/lm1b/input_pipeline.py has changed.
examples/ogbg_molpcba/models_test.py has changed.
tests/traverse_util_test.py has changed.
CHANGELOG.md has changed.
examples/wmt/tokenizer.py has changed.
examples/nlp_seq/train.py has changed.
examples/imagenet/imagenet_fake_data_benchmark.py has changed.
examples/lm1b/train_test.py has changed.
examples/ogbg_molpcba/configs/test.py has changed.
tests/core/core_frozen_dict_test.py has changed.
tests/tensorboard_test.py has changed.
examples/ogbg_molpcba/train_test.py has changed.
examples/ogbg_molpcba/configs/default.py has changed.
examples/sst2/models.py has changed.
examples/sst2/models_test.py has changed.
examples/imagenet/configs/default.py has changed.
...amples/imagenet/configs/v100_x8_mixed_precision.py has changed.
flax/metrics/__init__.py has changed.
docs/flax.linen.rst has changed.
examples/linen_design_test/linear_regression.py has changed.
docs/howtos/optax_update_guide.rst is new.
examples/seq2seq/requirements.txt has changed.
tests/traceback_util_test.py has changed.
tests/linen/linen_module_test.py has changed.
tests/linen/linen_linear_test.py has changed.
examples/ogbg_molpcba/requirements.txt has changed.
tests/linen/linen_transforms_test.py has changed.
examples/pixelcnn/train_test.py has changed.
examples/imagenet/configs/fake_data_benchmark.py has changed.
examples/ppo/ppo_lib_test.py has changed.
tests/core/design/core_tied_autoencoder_test.py has changed.
examples/wmt/decode.py has changed.
tests/early_stopping_test.py has changed.
examples/wmt/configs/default.py has changed.
examples/ppo/models.py has changed.
examples/wmt/train_test.py has changed.
examples/ppo/requirements.txt has changed.
examples/ppo/env_utils.py has changed.
tests/core/design/core_auto_encoder_test.py has changed.
examples/sst2/configs/default.py has changed.
examples/imagenet/train_test.py has changed.
examples/nlp_seq/requirements.txt is new.
examples/ogbg_molpcba/configs/hparam_sweep.py has changed.
examples/vae/utils.py has changed.
tests/nn_test.py has changed.
examples/linen_design_test/mlp_lazy.py has changed.
tests/linen/linen_attention_test.py has changed.
examples/mnist/requirements.txt has changed.
docs/howtos/lr_schedule.rst has changed.
examples/pixelcnn/input_pipeline.py has changed.
examples/ogbg_molpcba/input_pipeline_test.py has changed.
tests/nn_linear_test.py has changed.
tests/core/design/core_dense_test.py has changed.
examples/vae/requirements.txt has changed.
tests/linen/toplevel_test.py has changed.
examples/imagenet/configs/tpu.py has changed.
examples/lm1b/train.py has changed.
examples/imagenet/models.py has changed.
docs/contributing.md has changed.
tests/optim_test.py has changed.
examples/wmt/input_pipeline_test.py has changed.
tests/core/design/core_custom_vjp_test.py has changed.
examples/imagenet/configs/v100_x8.py has changed.
examples/pixelcnn/requirements.txt has changed.
examples/nlp_seq/models.py has changed.
tests/core/design/core_big_resnets_test.py has changed.
examples/imagenet/models_test.py has changed.
examples/imagenet/train.py has changed.
examples/wmt/requirements.txt has changed.
examples/linen_design_test/tied_autoencoder.py has changed.
tests/struct_test.py has changed.
tests/linen/linen_test.py has changed.
examples/ogbg_molpcba/configs/default_graph_net.py has changed.
examples/sst2/input_pipeline.py has changed.
examples/sst2/vocabulary.py 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/lm1b/input_pipeline_test.py has changed.
examples/ogbg_molpcba/ogbg_molpcba_benchmark.py has changed.
tests/core/design/core_resnet_test.py has changed.
examples/wmt/main.py has changed.
examples/wmt/input_pipeline.py has changed.
tests/checkpoints_test.py has changed.
examples/ogbg_molpcba/train.py has changed.
tests/serialization_test.py has changed.
examples/ogbg_molpcba/input_pipeline.py has changed.
examples/ppo/test_episodes.py has changed.
examples/linen_design_test/attention_simple.py has changed.
examples/lm1b/temperature_sampler.py has changed.
examples/pixelcnn/sample.py has changed.
examples/ppo/ppo_main.py has changed.
examples/mnist/train_test.py has changed.
examples/seq2seq/train.py has changed.
tests/core/design/core_vmap_test.py has changed.
tests/core/design/core_weight_std_test.py has changed.
dev/update_requirements.py is new.
examples/lm1b/tokenizer.py has changed.
examples/linen_design_test/dense.py has changed.
examples/mnist/configs/default.py has changed.
examples/pixelcnn/configs/default.py has changed.
examples/sst2/train_test.py has changed.
tests/core/design/core_scan_test.py has changed.
tests/linen/partitioning_test.py has changed.
examples/wmt/bleu.py has changed.
docs/notebooks/optax_update_guide.ipynb is new.
examples/mnist/mnist_benchmark.py has changed.
examples/mnist/train.py has changed.
tests/linen/dotgetter_test.py has changed.
examples/wmt/models.py has changed.
tests/nn_attention_test.py has changed.
examples/seq2seq/train_test.py has changed.
examples/cloud/launch_gce.py has changed.
examples/pixelcnn/model_test.py has changed.
examples/sst2/build_vocabulary.py has changed.
examples/imagenet/requirements.txt has changed.
flax/deprecated/__init__.py has changed.
examples/sst2/requirements.txt has changed.
examples/lm1b/configs/default.py has changed.
examples/nlp_seq/input_pipeline_test.py has changed.
tests/core/design/core_attention_test.py has changed.
examples/wmt/train.py has changed.
examples/ogbg_molpcba/models.py has changed.
examples/pixelcnn/main.py has changed.
docs/notebooks/annotated_mnist.ipynb has changed.
examples/sst2/input_pipeline_test.py has changed.
docs/index.rst has changed.

@@ -1,4 +1,4 @@
Loading
1 - 
# Copyright 2021 The Flax Authors.
1 + 
# Copyright 2022 The Flax Authors.
2 2 
#
3 3 
# Licensed under the Apache License, Version 2.0 (the "License");
4 4 
# you may not use this file except in compliance with the License.

@@ -1,4 +1,4 @@
Loading
1 - 
# Copyright 2021 The Flax Authors.
1 + 
# Copyright 2022 The Flax Authors.
2 2 
#
3 3 
# Licensed under the Apache License, Version 2.0 (the "License");
4 4 
# you may not use this file except in compliance with the License.

@@ -1,4 +1,4 @@
Loading
1 - 
# Copyright 2021 The Flax Authors.
1 + 
# Copyright 2022 The Flax Authors.
2 2 
#
3 3 
# Licensed under the Apache License, Version 2.0 (the "License");
4 4 
# you may not use this file except in compliance with the License.

@@ -1,4 +1,4 @@
Loading
1 - 
# Copyright 2021 The Flax Authors.
1 + 
# Copyright 2022 The Flax Authors.
2 2 
#
3 3 
# Licensed under the Apache License, Version 2.0 (the "License");
4 4 
# you may not use this file except in compliance with the License.

@@ -1,4 +1,4 @@
Loading
1 - 
# Copyright 2021 The Flax Authors.
1 + 
# Copyright 2022 The Flax Authors.
2 2 
#
3 3 
# Licensed under the Apache License, Version 2.0 (the "License");
4 4 
# you may not use this file except in compliance with the License.
@@ -14,20 +14,6 @@
Loading
14 14 










15
15
                             
                            
# Lint as: python 3










16
16
                             
                            










17
                            
                            -
                            
# Copyright 2020 The Flax Authors.









18
                            
                            -
                            
#









19
                            
                            -
                            
# Licensed under the Apache License, Version 2.0 (the "License");









20
                            
                            -
                            
# you may not use this file except in compliance with the License.









21
                            
                            -
                            
# You may obtain a copy of the License at









22
                            
                            -
                            
#









23
                            
                            -
                            
#      http://www.apache.org/licenses/LICENSE-2.0









24
                            
                            -
                            
#









25
                            
                            -
                            
# Unless required by applicable law or agreed to in writing, software









26
                            
                            -
                            
# distributed under the License is distributed on an "AS IS" BASIS,









27
                            
                            -
                            
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.









28
                            
                            -
                            
# See the License for the specific language governing permissions and









29
                            
                            -
                            
# limitations under the License.









30
                            
                            -
                            









31
17
                             
                            
"""Flax API."""










32
18
                             
                            










33
19
                             
                            
from . import core







 
 

















































































































































 










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...

















































































 










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...


































Everything is accounted for!

No changes detected that need to be reviewed.



What changes does Codecov check for?




Lines, not adjusted in diff, that have changed coverage data.


Files that introduced coverage data that had none before.


Files that have missing coverage data that once were tracked.









































51

Commits


















Anselm Levskaya
















Contextual



















Fix scope rewind-aliasing issue under transforms.











levskaya














64471a0






















Flax Authors
















Contextual



















Merge pull request #1723 from jheek:lazy-rng











a-googler














b60f7f4














Hiding 2 contexual commits
















Andreas Steiner


























83.35%



ø



ø


























Merge pull request #1774 from andsteing/optax











andsteing














42c9b60














































Andreas Steiner


























83.35%



ø





ø



























Applies Marc's remaining comments.











andsteing













#1774






8d4cf04





 CI Passed













































Andreas Steiner


























83.35%



ø



ø


























Merge pull request #1621 from andsteing/doc











andsteing














7c98266














































Andreas Steiner


























83.35%



ø






+0.35%




























Removed extra "test:".











andsteing














48a6f20





 CI Passed




























Andreas Steiner


























83.35%



ø






-0.02%




























Applied Marc's suggestions.











andsteing













#1774






a0369ae





 CI Passed


























-1







+1






















Marc van Zee


























83.37%



ø



ø


























Merge pull request #1793 from melissatan/patch-4











marcvanzee














9bb95f2














































Melissa Tan


























83.37%



ø





ø



























Fix typos.











melissatan













#1793






c907346





 CI Passed













































Material Components iOS Team


























83.37%



ø





ø



























Merge pull request #1784 from saeta:fix-license











a-googler














7d6844d





 CI Passed








































































83.37%



ø



ø


























Fix abstract PRNG shape when creating params.











jpuigcerver














03f685a









































































83.37%



ø





ø



























Fix duplicate licenses in other files too.











saeta













#1784






f835a55





 CI Passed








































































83.37%



ø





ø



























Remove duplicate license header.











saeta













#1784






4d83151





 CI Passed








































































83.37%



ø





ø



























Merge pull request #1699 from NeilGirdhar/patch-1











levskaya














89e1126





 CI Passed




























Andreas Steiner


























83.37%



ø





ø



























Two updates suggested by Avital.











andsteing













#1774






d17c0ae





 CI Passed




























Andreas Steiner


























83.37%



ø






-0.02%




























Update docs/howtos/optax_update_guide.rst











andsteing













#1774






a3cbbbc





 CI Passed




























Andreas Steiner


























83.37%



ø






-0.02%




























Update docs/howtos/optax_update_guide.rst











andsteing













#1774






aee1dba





 CI Passed






















































83.37%



ø






-0.02%




























Copybara import of the project:











Unknown














c1abf11





 CI Passed





















Flax Authors
















Contextual



















Merge pull request #1781 from google:dependabot/pip/examples/vae/pillow-9.0.0











a-googler














1500af6














Hiding 1 contexual commits




































Roy Frostig


























83.37%




75.00%







-0.02%




























validate RNG key shape against jax's default











froystig














c82cc74





 CI Passed

























+6






+4







+2
















































83.38%



ø





ø



























Bump pillow from 8.3.2 to 9.0.0 in /examples/vae











Unknown













#1781






2771a83





 CI Passed




























Andreas Steiner


























83.38%



ø





ø



























Adressing @rosshemsley's comments.











andsteing













#1774






057cdcd





 CI Passed




























Andreas Steiner


























83.38%



ø





ø



























Adds setup code and cross references LR schedules.











andsteing













#1774






e25260f





 CI Passed





















Andreas Steiner
















Contextual



















Adds optax update guide.











andsteing














1451cb8














Hiding 1 contexual commits































































83.38%




100.00%







+0.04%




























Handle setup with transformed methods taking submodules of self.











levskaya














fb4f2a4





 CI Passed

























+16






+16


















































83.34%



ø





ø



























Internal











Unknown














b36e873





 CI Passed













































Material Components iOS Team


























83.34%



ø



ø


























Merge pull request #1753 from andsteing:doc2











a-googler














63c5d5c









































































83.34%




100.00%







-0.04%




























revert setup transform changes out of caution, may not be generally safe.











levskaya














6a2e1b3





 CI Passed

























-14






-14
























Material Components iOS Team


























83.38%




100.00%







+0.04%




























Merge pull request #1745 from levskaya:setup_idempotency











a-googler














712e780





 CI Passed

























+14






+14
























Andreas Steiner


























83.34%



ø



ø


























Makes `FLAX_PROFILE=true` gotcha glaringly obvious.











andsteing













#1753






24ce961








































































83.34%



ø





ø



























Copybara import of the project:











Unknown














151dbc1





 CI Passed





















Flax Authors
















Contextual



















Merge pull request #1749 from google:dependabot/pip/examples/pixelcnn/pillow-8.3.2











a-googler














830013c














Hiding 1 contexual commits






























































83.34%



ø



ø


























Bump pillow from 8.2.0 to 8.3.2 in /examples/pixelcnn











Unknown













#1749






fb97474








































































83.34%



ø



ø


























Internal











Unknown














9775f56














































Material Components iOS Team


























83.34%



ø



ø


























Merge pull request #1726 from andsteing:examples











a-googler














373aecd


























-14






-14



















































83.38%




100.00%







+0.04%




























Handle setup with transformed methods taking submodules of self.











levskaya













#1745






baf8015





 CI Passed

























+14






+14
























Material Components iOS Team


























83.34%




100.00%







+0.03%




























Merge pull request #1738 from jheek:custom-vjp











a-googler














25dd368





 CI Passed

























+11






+11
























Material Components iOS Team


























83.31%




100.00%







+0.06%




























Merge pull request #1739 from jheek:param_dtype











a-googler














9639ed7





 CI Passed

























+10






+10



















































83.28%




100.00%







+0.05%




























Implement custom vjp











jheek













#1738






ca4fb9a





 CI Passed

























-10






-10



















































83.31%



ø



ø


























Add param_dtype argument to linen Modules











jheek













#1739






b67a0d0


























+21






+21
























Marc van Zee


























83.25%



ø






-0.01%




























Update normalization.py











marcvanzee














3e9c8f5





 CI Passed













































Andreas Steiner


























83.25%



ø



ø


























Makes script use `flags.mark_flag_as_required()`.











andsteing













#1726






8352fcc





























Andreas Steiner


























83.25%



ø



ø


























Improves script `dev/update_requirements.py`.











andsteing













#1726






381bd03














































Andreas Steiner


























83.25%



ø






-0.01%




























Fixes numpy warning.











andsteing













#1726






25ae2c0





 CI Passed





















Andreas Steiner
















Contextual



















Adds env/update_requirements.py and runs it.











andsteing














086fc97






















Andreas Steiner
















Contextual



















Updates requirements.txt & re-orders imports.











andsteing














ff9e9eb














Hiding 2 contexual commits




































Marc van Zee


























83.25%



ø



ø


























Rolling back previous change due to internal breakage.











marcvanzee














4acd0fe


























+24






+24



















































83.18%



ø



ø


























Remove unnecessary dependency











NeilGirdhar













#1699






e8cf91a


























+81






+78







+3






















Andreas Steiner


























83.00%



ø



ø


























Removes backticks around Module.apply











andsteing














5e6b22f


























+3






+3
























Andreas Steiner


























82.99%



ø



ø


























Fixes RST link formatting.











andsteing














680abcb


























-142






-126







-16

















































83.12%




25.00%







-0.16%




























Put lazy rng behind a flag











jheek













#1723






a591f96





 CI Passed








































































Files






Coverage
























flax



















0.05%






83.17%











Project Totals (69 files)






83.17%
































Loading