diff --git a/examples/imagenet/configs/default.py b/examples/imagenet/configs/default.py index a105bfdac2..c90d7a5972 100644 --- a/examples/imagenet/configs/default.py +++ b/examples/imagenet/configs/default.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2021 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/examples/imagenet/configs/tpu.py b/examples/imagenet/configs/tpu.py index 253031b681..ae4e9dc8d4 100644 --- a/examples/imagenet/configs/tpu.py +++ b/examples/imagenet/configs/tpu.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2021 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/examples/sst2/build_vocabulary.py b/examples/sst2/build_vocabulary.py old mode 100755 new mode 100644 diff --git a/examples/sst2/input_pipeline.py b/examples/sst2/input_pipeline.py old mode 100755 new mode 100644 diff --git a/examples/sst2/vocabulary.py b/examples/sst2/vocabulary.py old mode 100755 new mode 100644 diff --git a/examples/vae/train.py b/examples/vae/train.py index 72895a23f3..84f1b582a0 100644 --- a/examples/vae/train.py +++ b/examples/vae/train.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2023 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/examples/vae/utils.py b/examples/vae/utils.py index 317655b635..2881141994 100644 --- a/examples/vae/utils.py +++ b/examples/vae/utils.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2023 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/linen/__init__.py b/flax/linen/__init__.py index 9f0bfc84b4..9b80ca3c18 100644 --- a/flax/linen/__init__.py +++ b/flax/linen/__init__.py @@ -12,20 +12,6 @@ # See the License for the specific language governing permissions and # limitations under the License. -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - """The Flax Module system.""" diff --git a/flax/linen/batch_apply.py b/flax/linen/batch_apply.py index 6f45f016aa..780f18e163 100644 --- a/flax/linen/batch_apply.py +++ b/flax/linen/batch_apply.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2023 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/linen/dtypes.py b/flax/linen/dtypes.py index d88f998180..467ab89dc9 100644 --- a/flax/linen/dtypes.py +++ b/flax/linen/dtypes.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2022 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/linen/transforms.py b/flax/linen/transforms.py index db87512d43..1e23dfefd8 100644 --- a/flax/linen/transforms.py +++ b/flax/linen/transforms.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2023 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/metrics/__init__.py b/flax/metrics/__init__.py index c562845e66..e80ba0b35f 100644 --- a/flax/metrics/__init__.py +++ b/flax/metrics/__init__.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2023 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/examples/gemma/helpers.py b/flax/nnx/examples/gemma/helpers.py index 37cc800b7f..1797f01271 100644 --- a/flax/nnx/examples/gemma/helpers.py +++ b/flax/nnx/examples/gemma/helpers.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/examples/gemma/helpers_test.py b/flax/nnx/examples/gemma/helpers_test.py index 2bac0ab8c2..8d5e899f93 100644 --- a/flax/nnx/examples/gemma/helpers_test.py +++ b/flax/nnx/examples/gemma/helpers_test.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/examples/gemma/layers.py b/flax/nnx/examples/gemma/layers.py index 241bbba52a..174ec7272c 100644 --- a/flax/nnx/examples/gemma/layers.py +++ b/flax/nnx/examples/gemma/layers.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/examples/gemma/layers_test.py b/flax/nnx/examples/gemma/layers_test.py index 08c42248ea..d71071bdbf 100644 --- a/flax/nnx/examples/gemma/layers_test.py +++ b/flax/nnx/examples/gemma/layers_test.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/examples/gemma/modules.py b/flax/nnx/examples/gemma/modules.py index bf62faa465..d09939491e 100644 --- a/flax/nnx/examples/gemma/modules.py +++ b/flax/nnx/examples/gemma/modules.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/examples/gemma/modules_test.py b/flax/nnx/examples/gemma/modules_test.py index b7f6454ec8..b585cb3c53 100644 --- a/flax/nnx/examples/gemma/modules_test.py +++ b/flax/nnx/examples/gemma/modules_test.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/examples/gemma/params.py b/flax/nnx/examples/gemma/params.py index da3b6ec5fb..a05ae8fc71 100644 --- a/flax/nnx/examples/gemma/params.py +++ b/flax/nnx/examples/gemma/params.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/examples/gemma/positional_embeddings.py b/flax/nnx/examples/gemma/positional_embeddings.py index b0ed75f4a0..4fa77b88a6 100644 --- a/flax/nnx/examples/gemma/positional_embeddings.py +++ b/flax/nnx/examples/gemma/positional_embeddings.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/examples/gemma/positional_embeddings_test.py b/flax/nnx/examples/gemma/positional_embeddings_test.py index 421a1570b4..3a85790440 100644 --- a/flax/nnx/examples/gemma/positional_embeddings_test.py +++ b/flax/nnx/examples/gemma/positional_embeddings_test.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/examples/gemma/sampler.py b/flax/nnx/examples/gemma/sampler.py index 5fd065303b..d1f4f26c4e 100644 --- a/flax/nnx/examples/gemma/sampler.py +++ b/flax/nnx/examples/gemma/sampler.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/examples/gemma/sampler_test.py b/flax/nnx/examples/gemma/sampler_test.py index e3993a633e..2e131dda74 100644 --- a/flax/nnx/examples/gemma/sampler_test.py +++ b/flax/nnx/examples/gemma/sampler_test.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/examples/gemma/transformer.py b/flax/nnx/examples/gemma/transformer.py index bba583e46d..b4543316e5 100644 --- a/flax/nnx/examples/gemma/transformer.py +++ b/flax/nnx/examples/gemma/transformer.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/examples/gemma/transformer_test.py b/flax/nnx/examples/gemma/transformer_test.py index e9628660a5..b2c852af4e 100644 --- a/flax/nnx/examples/gemma/transformer_test.py +++ b/flax/nnx/examples/gemma/transformer_test.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/nnx/__init__.py b/flax/nnx/nnx/__init__.py index fcd485307b..a766d9b4e6 100644 --- a/flax/nnx/nnx/__init__.py +++ b/flax/nnx/nnx/__init__.py @@ -11,17 +11,3 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. - -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. diff --git a/flax/nnx/nnx/helpers.py b/flax/nnx/nnx/helpers.py index fe2ad8daf3..28e94a3660 100644 --- a/flax/nnx/nnx/helpers.py +++ b/flax/nnx/nnx/helpers.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/nnx/ids.py b/flax/nnx/nnx/ids.py index d72fae3e22..7e51179edd 100644 --- a/flax/nnx/nnx/ids.py +++ b/flax/nnx/nnx/ids.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/nnx/nn/__init__.py b/flax/nnx/nnx/nn/__init__.py index fcd485307b..a766d9b4e6 100644 --- a/flax/nnx/nnx/nn/__init__.py +++ b/flax/nnx/nnx/nn/__init__.py @@ -11,17 +11,3 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. - -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. diff --git a/flax/nnx/nnx/nn/linear.py b/flax/nnx/nnx/nn/linear.py index c3e72982fd..8eaa6b124b 100644 --- a/flax/nnx/nnx/nn/linear.py +++ b/flax/nnx/nnx/nn/linear.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/nnx/nn/lora.py b/flax/nnx/nnx/nn/lora.py index 740357e887..dccc8aaffd 100644 --- a/flax/nnx/nnx/nn/lora.py +++ b/flax/nnx/nnx/nn/lora.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/nnx/nn/stochastic.py b/flax/nnx/nnx/nn/stochastic.py index 93de8771f9..0a645cb08b 100644 --- a/flax/nnx/nnx/nn/stochastic.py +++ b/flax/nnx/nnx/nn/stochastic.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/nnx/proxy_caller.py b/flax/nnx/nnx/proxy_caller.py index 7fe872a4ad..e0b5e91892 100644 --- a/flax/nnx/nnx/proxy_caller.py +++ b/flax/nnx/nnx/proxy_caller.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/nnx/rnglib.py b/flax/nnx/nnx/rnglib.py index 6b2520f4ac..c9e97c0d13 100644 --- a/flax/nnx/nnx/rnglib.py +++ b/flax/nnx/nnx/rnglib.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/nnx/state.py b/flax/nnx/nnx/state.py index 32e60cb3be..0949d8e8c1 100644 --- a/flax/nnx/nnx/state.py +++ b/flax/nnx/nnx/state.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/nnx/training/metrics.py b/flax/nnx/nnx/training/metrics.py index a20d500904..f17485ea31 100644 --- a/flax/nnx/nnx/training/metrics.py +++ b/flax/nnx/nnx/training/metrics.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/nnx/training/optimizer.py b/flax/nnx/nnx/training/optimizer.py index a8f5fae9f0..cd5fffe62c 100644 --- a/flax/nnx/nnx/training/optimizer.py +++ b/flax/nnx/nnx/training/optimizer.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/nnx/transforms/compilation.py b/flax/nnx/nnx/transforms/compilation.py index 766f25ec81..3ca8fdf94a 100644 --- a/flax/nnx/nnx/transforms/compilation.py +++ b/flax/nnx/nnx/transforms/compilation.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/nnx/transforms/deprecated.py b/flax/nnx/nnx/transforms/deprecated.py index 3dd826b254..a1f8c82e5d 100644 --- a/flax/nnx/nnx/transforms/deprecated.py +++ b/flax/nnx/nnx/transforms/deprecated.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/nnx/transforms/general.py b/flax/nnx/nnx/transforms/general.py index 93a328f4eb..f86afa9576 100644 --- a/flax/nnx/nnx/transforms/general.py +++ b/flax/nnx/nnx/transforms/general.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/nnx/transforms/iteration.py b/flax/nnx/nnx/transforms/iteration.py index 4315f1a379..8c242d5db6 100644 --- a/flax/nnx/nnx/transforms/iteration.py +++ b/flax/nnx/nnx/transforms/iteration.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/nnx/transforms/transforms.py b/flax/nnx/nnx/transforms/transforms.py index 403fb2af66..750190bebd 100644 --- a/flax/nnx/nnx/transforms/transforms.py +++ b/flax/nnx/nnx/transforms/transforms.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/nnx/variables.py b/flax/nnx/nnx/variables.py index 1348cd4687..c4cb5ec9de 100644 --- a/flax/nnx/nnx/variables.py +++ b/flax/nnx/nnx/variables.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2024 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/flax/nnx/tests/__init__.py b/flax/nnx/tests/__init__.py index fcd485307b..a766d9b4e6 100644 --- a/flax/nnx/tests/__init__.py +++ b/flax/nnx/tests/__init__.py @@ -11,17 +11,3 @@ # WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. - -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. diff --git a/tests/linen/linen_batch_apply_test.py b/tests/linen/linen_batch_apply_test.py index 7a53ac16f4..dedc0d3f07 100644 --- a/tests/linen/linen_batch_apply_test.py +++ b/tests/linen/linen_batch_apply_test.py @@ -1,17 +1,3 @@ -# Copyright 2024 The Flax Authors. -# -# Licensed under the Apache License, Version 2.0 (the "License"); -# you may not use this file except in compliance with the License. -# You may obtain a copy of the License at -# -# http://www.apache.org/licenses/LICENSE-2.0 -# -# Unless required by applicable law or agreed to in writing, software -# distributed under the License is distributed on an "AS IS" BASIS, -# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. -# See the License for the specific language governing permissions and -# limitations under the License. - # Copyright 2023 The Flax Authors. # # Licensed under the Apache License, Version 2.0 (the "License");