diff --git a/.github/workflows/cat_slurm_logs.py b/.github/workflows/cat_slurm_logs.py index 50e9112843d5..0479a4723302 100644 --- a/.github/workflows/cat_slurm_logs.py +++ b/.github/workflows/cat_slurm_logs.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/.github/workflows/parse_logs.py b/.github/workflows/parse_logs.py index 7cf7454520d8..d2290a6080aa 100644 --- a/.github/workflows/parse_logs.py +++ b/.github/workflows/parse_logs.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/.github/workflows/slurm_job_scripts/slurm_utils_common.sh b/.github/workflows/slurm_job_scripts/slurm_utils_common.sh index 030e9d74785c..93dd1bf0b014 100755 --- a/.github/workflows/slurm_job_scripts/slurm_utils_common.sh +++ b/.github/workflows/slurm_job_scripts/slurm_utils_common.sh @@ -1,6 +1,6 @@ #! /bin/bash -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/AUTHORS b/AUTHORS new file mode 100644 index 000000000000..7281acf41a9a --- /dev/null +++ b/AUTHORS @@ -0,0 +1,8 @@ +# This is the list of JAX's significant contributors. +# +# This does not necessarily list everyone who has contributed code, +# especially since many employees of one corporation may be contributing. +# To see the full list of contributors, see the revision history in +# source control. +Google LLC +NVIDIA Corporation \ No newline at end of file diff --git a/benchmarks/api_benchmark.py b/benchmarks/api_benchmark.py index 32e4fe059f55..60ababad1e1f 100644 --- a/benchmarks/api_benchmark.py +++ b/benchmarks/api_benchmark.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/benchmarks/benchmark.py b/benchmarks/benchmark.py index 0195468d11ed..f066c6fc28ea 100644 --- a/benchmarks/benchmark.py +++ b/benchmarks/benchmark.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/benchmarks/gda_benchmark.py b/benchmarks/gda_benchmark.py index 726e00d97fbc..ffe7f160e686 100644 --- a/benchmarks/gda_benchmark.py +++ b/benchmarks/gda_benchmark.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/benchmarks/pmap_benchmark.py b/benchmarks/pmap_benchmark.py index a0f52d8c5d33..5b400a75ec57 100644 --- a/benchmarks/pmap_benchmark.py +++ b/benchmarks/pmap_benchmark.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/build/BUILD.bazel b/build/BUILD.bazel index 8e8118a4f3d3..889f5a730872 100644 --- a/build/BUILD.bazel +++ b/build/BUILD.bazel @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/build/build.py b/build/build.py index c37d57153242..86058053254a 100755 --- a/build/build.py +++ b/build/build.py @@ -1,6 +1,6 @@ #!/usr/bin/python # -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/build/build_wheel.py b/build/build_wheel.py index 75df869f3543..f10efab99997 100644 --- a/build/build_wheel.py +++ b/build/build_wheel.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/build/rocm/build_common.sh b/build/rocm/build_common.sh index ef4f7896f229..eb7a3225f352 100644 --- a/build/rocm/build_common.sh +++ b/build/rocm/build_common.sh @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/build/rocm/build_rocm.sh b/build/rocm/build_rocm.sh index 49c530ac5999..559368c73314 100755 --- a/build/rocm/build_rocm.sh +++ b/build/rocm/build_rocm.sh @@ -1,5 +1,5 @@ #!/usr/bin/env bash -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/build/rocm/ci_build.sh b/build/rocm/ci_build.sh index 16002a484fbc..1c081e004e24 100755 --- a/build/rocm/ci_build.sh +++ b/build/rocm/ci_build.sh @@ -1,5 +1,5 @@ #!/usr/bin/env bash -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/build/rocm/run_multi_gpu.sh b/build/rocm/run_multi_gpu.sh index 49ee3e88b1d5..caad6f1a585f 100755 --- a/build/rocm/run_multi_gpu.sh +++ b/build/rocm/run_multi_gpu.sh @@ -1,5 +1,5 @@ #!/usr/bin/env bash -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/build/rocm/run_single_gpu.py b/build/rocm/run_single_gpu.py index 4db584302283..ebb65a31745b 100755 --- a/build/rocm/run_single_gpu.py +++ b/build/rocm/run_single_gpu.py @@ -1,5 +1,5 @@ #!/usr/bin/env python3 -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/conftest.py b/conftest.py index c33e7f88bd71..dbc1926fb865 100644 --- a/conftest.py +++ b/conftest.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/docs/autodidax.ipynb b/docs/autodidax.ipynb index 7a26ce5985dd..a4b284eebe58 100644 --- a/docs/autodidax.ipynb +++ b/docs/autodidax.ipynb @@ -5,7 +5,7 @@ "metadata": {}, "source": [ "---\n", - "Copyright 2021 Google LLC\n", + "Copyright 2021 The JAX Authors.\n", "\n", "Licensed under the Apache License, Version 2.0 (the \"License\");\n", "you may not use this file except in compliance with the License.\n", diff --git a/docs/autodidax.md b/docs/autodidax.md index 8fe47592edd2..47d02f6f0399 100644 --- a/docs/autodidax.md +++ b/docs/autodidax.md @@ -15,7 +15,7 @@ kernelspec: ```{raw-cell} --- -Copyright 2021 Google LLC +Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/docs/autodidax.py b/docs/autodidax.py index 7fa84f82d784..c8d50c641f1b 100644 --- a/docs/autodidax.py +++ b/docs/autodidax.py @@ -1,5 +1,5 @@ # --- -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/docs/conf.py b/docs/conf.py index 1f27cd6d1cc5..78e0be7f2628 100644 --- a/docs/conf.py +++ b/docs/conf.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. @@ -40,7 +40,7 @@ # -- Project information ----------------------------------------------------- project = 'JAX' -copyright = '2020, Google LLC. NumPy and SciPy documentation are copyright the respective authors.' +copyright = '2020, The JAX Authors. NumPy and SciPy documentation are copyright the respective authors.' author = 'The JAX authors' # The short X.Y version diff --git a/docs/notebooks/Neural_Network_and_Data_Loading.ipynb b/docs/notebooks/Neural_Network_and_Data_Loading.ipynb index 6824a204879a..40f8a175fcac 100644 --- a/docs/notebooks/Neural_Network_and_Data_Loading.ipynb +++ b/docs/notebooks/Neural_Network_and_Data_Loading.ipynb @@ -10,7 +10,7 @@ "\n", "[![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/jax/blob/main/docs/notebooks/Neural_Network_and_Data_Loading.ipynb)\n", "\n", - "**Copyright 2018 Google LLC.**\n", + "**Copyright 2018 The JAX Authors.**\n", "\n", "Licensed under the Apache License, Version 2.0 (the \"License\"); you may not use this file except in compliance with the License.\n", "You may obtain a copy of the License at\n", diff --git a/docs/notebooks/Neural_Network_and_Data_Loading.md b/docs/notebooks/Neural_Network_and_Data_Loading.md index 74d15fc78046..0241b731b65d 100644 --- a/docs/notebooks/Neural_Network_and_Data_Loading.md +++ b/docs/notebooks/Neural_Network_and_Data_Loading.md @@ -18,7 +18,7 @@ kernelspec: [![Open in Colab](https://colab.research.google.com/assets/colab-badge.svg)](https://colab.research.google.com/github/google/jax/blob/main/docs/notebooks/Neural_Network_and_Data_Loading.ipynb) -**Copyright 2018 Google LLC.** +**Copyright 2018 The JAX 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 diff --git a/docs/notebooks/neural_network_with_tfds_data.ipynb b/docs/notebooks/neural_network_with_tfds_data.ipynb index 00e129969f3b..421fc8e712b8 100644 --- a/docs/notebooks/neural_network_with_tfds_data.ipynb +++ b/docs/notebooks/neural_network_with_tfds_data.ipynb @@ -6,7 +6,7 @@ "id": "18AF5Ab4p6VL" }, "source": [ - "**Copyright 2018 Google LLC.**\n", + "**Copyright 2018 The JAX Authors.**\n", "\n", "Licensed under the Apache License, Version 2.0 (the \"License\");" ] diff --git a/docs/notebooks/neural_network_with_tfds_data.md b/docs/notebooks/neural_network_with_tfds_data.md index 1492956e5790..b29a7754b799 100644 --- a/docs/notebooks/neural_network_with_tfds_data.md +++ b/docs/notebooks/neural_network_with_tfds_data.md @@ -14,7 +14,7 @@ kernelspec: +++ {"id": "18AF5Ab4p6VL"} -**Copyright 2018 Google LLC.** +**Copyright 2018 The JAX Authors.** Licensed under the Apache License, Version 2.0 (the "License"); diff --git a/docs/sphinxext/jax_extensions.py b/docs/sphinxext/jax_extensions.py index e02f12215f28..4e479eec37d4 100644 --- a/docs/sphinxext/jax_extensions.py +++ b/docs/sphinxext/jax_extensions.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/examples/__init__.py b/examples/__init__.py index b0c7da3d7725..0a0a8260408e 100644 --- a/examples/__init__.py +++ b/examples/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/examples/advi.py b/examples/advi.py index 8c5cfe14b486..35ee94a58d2f 100644 --- a/examples/advi.py +++ b/examples/advi.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/examples/datasets.py b/examples/datasets.py index 3a482132dad2..e6b1980c0e81 100644 --- a/examples/datasets.py +++ b/examples/datasets.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/examples/differentially_private_sgd.py b/examples/differentially_private_sgd.py index 6e5cf253938d..c008d49ef74c 100644 --- a/examples/differentially_private_sgd.py +++ b/examples/differentially_private_sgd.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/examples/examples_test.py b/examples/examples_test.py index b3407b3593d1..ce829c5fbf59 100644 --- a/examples/examples_test.py +++ b/examples/examples_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/examples/gaussian_process_regression.py b/examples/gaussian_process_regression.py index 479194563504..e14b57812260 100644 --- a/examples/gaussian_process_regression.py +++ b/examples/gaussian_process_regression.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/examples/jax_cpp/BUILD b/examples/jax_cpp/BUILD index bc7fa9b8b6d3..7c5c12ff4fff 100644 --- a/examples/jax_cpp/BUILD +++ b/examples/jax_cpp/BUILD @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/examples/jax_cpp/main.cc b/examples/jax_cpp/main.cc index 731dee31e6c1..cbc2ffdfe9b7 100644 --- a/examples/jax_cpp/main.cc +++ b/examples/jax_cpp/main.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/examples/jax_cpp/prog.py b/examples/jax_cpp/prog.py index 8a17adce432f..77be39d83ebf 100644 --- a/examples/jax_cpp/prog.py +++ b/examples/jax_cpp/prog.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/examples/kernel_lsq.py b/examples/kernel_lsq.py index 59521f87b64f..472ebe82c8a7 100644 --- a/examples/kernel_lsq.py +++ b/examples/kernel_lsq.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/examples/mnist_classifier.py b/examples/mnist_classifier.py index 3d9ca4752a98..a0fe8b996c98 100644 --- a/examples/mnist_classifier.py +++ b/examples/mnist_classifier.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/examples/mnist_classifier_fromscratch.py b/examples/mnist_classifier_fromscratch.py index cc88086e7e7e..911126799d6d 100644 --- a/examples/mnist_classifier_fromscratch.py +++ b/examples/mnist_classifier_fromscratch.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/examples/mnist_vae.py b/examples/mnist_vae.py index f05858ef44f9..141be978f635 100644 --- a/examples/mnist_vae.py +++ b/examples/mnist_vae.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/examples/onnx2xla.py b/examples/onnx2xla.py index 196ab1c03b53..0a9494e3dea2 100644 --- a/examples/onnx2xla.py +++ b/examples/onnx2xla.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/examples/spmd_mnist_classifier_fromscratch.py b/examples/spmd_mnist_classifier_fromscratch.py index bb3e085266da..3698314708c7 100644 --- a/examples/spmd_mnist_classifier_fromscratch.py +++ b/examples/spmd_mnist_classifier_fromscratch.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/BUILD b/jax/BUILD index ad5b3bf1d31f..70df4fb3de7a 100644 --- a/jax/BUILD +++ b/jax/BUILD @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/__init__.py b/jax/__init__.py index d7ac714fd58e..92316b42e84b 100644 --- a/jax/__init__.py +++ b/jax/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/__init__.py b/jax/_src/__init__.py index 30257881ab32..60afaf5a8da1 100644 --- a/jax/_src/__init__.py +++ b/jax/_src/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/abstract_arrays.py b/jax/_src/abstract_arrays.py index 577bdaea4c1a..15d7049f1848 100644 --- a/jax/_src/abstract_arrays.py +++ b/jax/_src/abstract_arrays.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/ad_checkpoint.py b/jax/_src/ad_checkpoint.py index 54cf7246b59c..616503f073ef 100644 --- a/jax/_src/ad_checkpoint.py +++ b/jax/_src/ad_checkpoint.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/ad_util.py b/jax/_src/ad_util.py index e9f9eb24b3b6..86488e11e7a1 100644 --- a/jax/_src/ad_util.py +++ b/jax/_src/ad_util.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/api.py b/jax/_src/api.py index 363641946a60..3b07f721627e 100644 --- a/jax/_src/api.py +++ b/jax/_src/api.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/api_util.py b/jax/_src/api_util.py index 62127fa4974b..fe4f33aef11e 100644 --- a/jax/_src/api_util.py +++ b/jax/_src/api_util.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/callback.py b/jax/_src/callback.py index ced20d119da5..2ab088dc768a 100644 --- a/jax/_src/callback.py +++ b/jax/_src/callback.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/checkify.py b/jax/_src/checkify.py index 5a48a14c18b6..679bc8c04c08 100644 --- a/jax/_src/checkify.py +++ b/jax/_src/checkify.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/cloud_tpu_init.py b/jax/_src/cloud_tpu_init.py index db82c4888f9f..908f688d3f99 100644 --- a/jax/_src/cloud_tpu_init.py +++ b/jax/_src/cloud_tpu_init.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/config.py b/jax/_src/config.py index 37144ade9561..64e55084c2d4 100644 --- a/jax/_src/config.py +++ b/jax/_src/config.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/custom_api_util.py b/jax/_src/custom_api_util.py index 5fe40937afdc..76b421f37403 100644 --- a/jax/_src/custom_api_util.py +++ b/jax/_src/custom_api_util.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/custom_batching.py b/jax/_src/custom_batching.py index aa639a00e607..2cc933f00da6 100644 --- a/jax/_src/custom_batching.py +++ b/jax/_src/custom_batching.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/custom_derivatives.py b/jax/_src/custom_derivatives.py index 59d84c23698e..36b05fd8b96b 100644 --- a/jax/_src/custom_derivatives.py +++ b/jax/_src/custom_derivatives.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/custom_transpose.py b/jax/_src/custom_transpose.py index 5a3bff057357..3ff980dde48c 100644 --- a/jax/_src/custom_transpose.py +++ b/jax/_src/custom_transpose.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/debugger/__init__.py b/jax/_src/debugger/__init__.py index e2e6b04749c7..5e367d3fc352 100644 --- a/jax/_src/debugger/__init__.py +++ b/jax/_src/debugger/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/debugger/cli_debugger.py b/jax/_src/debugger/cli_debugger.py index 9c53ecfec9a9..498a975de0c7 100644 --- a/jax/_src/debugger/cli_debugger.py +++ b/jax/_src/debugger/cli_debugger.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/debugger/colab_debugger.py b/jax/_src/debugger/colab_debugger.py index f200f0abdd18..12588d7568e3 100644 --- a/jax/_src/debugger/colab_debugger.py +++ b/jax/_src/debugger/colab_debugger.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/debugger/colab_lib.py b/jax/_src/debugger/colab_lib.py index ec823d271a6e..42a3d73efb5e 100644 --- a/jax/_src/debugger/colab_lib.py +++ b/jax/_src/debugger/colab_lib.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/debugger/core.py b/jax/_src/debugger/core.py index 5a8f0e109401..bfa29ed64331 100644 --- a/jax/_src/debugger/core.py +++ b/jax/_src/debugger/core.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/debugger/web_debugger.py b/jax/_src/debugger/web_debugger.py index 56728095f21d..9f12c19bb875 100644 --- a/jax/_src/debugger/web_debugger.py +++ b/jax/_src/debugger/web_debugger.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/debugging.py b/jax/_src/debugging.py index c88ecf463d61..39df27280f59 100644 --- a/jax/_src/debugging.py +++ b/jax/_src/debugging.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/device_array.py b/jax/_src/device_array.py index 58982416cc70..3214be054f81 100644 --- a/jax/_src/device_array.py +++ b/jax/_src/device_array.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/dispatch.py b/jax/_src/dispatch.py index 4bcbebd66539..273460e323cb 100644 --- a/jax/_src/dispatch.py +++ b/jax/_src/dispatch.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/distributed.py b/jax/_src/distributed.py index d4c96dd37d1e..24fa573f98cb 100644 --- a/jax/_src/distributed.py +++ b/jax/_src/distributed.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/dlpack.py b/jax/_src/dlpack.py index 3f45995a4222..d604ecf9df0d 100644 --- a/jax/_src/dlpack.py +++ b/jax/_src/dlpack.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/dtypes.py b/jax/_src/dtypes.py index 894b6247d649..8d009c472fc7 100644 --- a/jax/_src/dtypes.py +++ b/jax/_src/dtypes.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/environment_info.py b/jax/_src/environment_info.py index 5cfc6a3d02e6..2c1c07d3dc1a 100644 --- a/jax/_src/environment_info.py +++ b/jax/_src/environment_info.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/errors.py b/jax/_src/errors.py index b8359ebd6af0..78a17938f37e 100644 --- a/jax/_src/errors.py +++ b/jax/_src/errors.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/flatten_util.py b/jax/_src/flatten_util.py index f85367db6b48..bc974966fe06 100644 --- a/jax/_src/flatten_util.py +++ b/jax/_src/flatten_util.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/image/__init__.py b/jax/_src/image/__init__.py index 30257881ab32..60afaf5a8da1 100644 --- a/jax/_src/image/__init__.py +++ b/jax/_src/image/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/image/scale.py b/jax/_src/image/scale.py index 4cf4f7d8eb2b..5a2946b9e3ab 100644 --- a/jax/_src/image/scale.py +++ b/jax/_src/image/scale.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/iree.py b/jax/_src/iree.py index d1316e795ef1..1af3f0c20e83 100644 --- a/jax/_src/iree.py +++ b/jax/_src/iree.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/__init__.py b/jax/_src/lax/__init__.py index b6bf557f2c4d..a06eea6941cb 100644 --- a/jax/_src/lax/__init__.py +++ b/jax/_src/lax/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/ann.py b/jax/_src/lax/ann.py index 74223f090230..f1d010362fcc 100644 --- a/jax/_src/lax/ann.py +++ b/jax/_src/lax/ann.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/control_flow/__init__.py b/jax/_src/lax/control_flow/__init__.py index fc2efd747c2c..4cf320d93f13 100644 --- a/jax/_src/lax/control_flow/__init__.py +++ b/jax/_src/lax/control_flow/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/control_flow/common.py b/jax/_src/lax/control_flow/common.py index 05b6d705b9d9..6af6c666aed1 100644 --- a/jax/_src/lax/control_flow/common.py +++ b/jax/_src/lax/control_flow/common.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/control_flow/conditionals.py b/jax/_src/lax/control_flow/conditionals.py index d466984ff881..ed82f9807673 100644 --- a/jax/_src/lax/control_flow/conditionals.py +++ b/jax/_src/lax/control_flow/conditionals.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/control_flow/for_loop.py b/jax/_src/lax/control_flow/for_loop.py index bd35ce210ea4..22418eaaf0b4 100644 --- a/jax/_src/lax/control_flow/for_loop.py +++ b/jax/_src/lax/control_flow/for_loop.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/control_flow/loops.py b/jax/_src/lax/control_flow/loops.py index 61e45605d8cf..a5a8080c1d47 100644 --- a/jax/_src/lax/control_flow/loops.py +++ b/jax/_src/lax/control_flow/loops.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/control_flow/solves.py b/jax/_src/lax/control_flow/solves.py index 3cfa8d3a427c..a4e02322b7ad 100644 --- a/jax/_src/lax/control_flow/solves.py +++ b/jax/_src/lax/control_flow/solves.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/convolution.py b/jax/_src/lax/convolution.py index dc03a4602e32..57e534aa12fb 100644 --- a/jax/_src/lax/convolution.py +++ b/jax/_src/lax/convolution.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/eigh.py b/jax/_src/lax/eigh.py index d1c7fe9b43af..76e9d554937f 100644 --- a/jax/_src/lax/eigh.py +++ b/jax/_src/lax/eigh.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/fft.py b/jax/_src/lax/fft.py index 5a28569b5a43..609e072e18b0 100644 --- a/jax/_src/lax/fft.py +++ b/jax/_src/lax/fft.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/lax.py b/jax/_src/lax/lax.py index 1a2ddab06c96..335416486911 100644 --- a/jax/_src/lax/lax.py +++ b/jax/_src/lax/lax.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/linalg.py b/jax/_src/lax/linalg.py index 2555ad2095f5..c7d398f11298 100644 --- a/jax/_src/lax/linalg.py +++ b/jax/_src/lax/linalg.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/other.py b/jax/_src/lax/other.py index d95cfadb0501..f3eab7efcddc 100644 --- a/jax/_src/lax/other.py +++ b/jax/_src/lax/other.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/parallel.py b/jax/_src/lax/parallel.py index be87c1e5a20f..47babc4db4b3 100644 --- a/jax/_src/lax/parallel.py +++ b/jax/_src/lax/parallel.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/qdwh.py b/jax/_src/lax/qdwh.py index 11533c3a5499..9750a7c9469b 100644 --- a/jax/_src/lax/qdwh.py +++ b/jax/_src/lax/qdwh.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/slicing.py b/jax/_src/lax/slicing.py index 35165d85ed26..4ed84a684588 100644 --- a/jax/_src/lax/slicing.py +++ b/jax/_src/lax/slicing.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/stack.py b/jax/_src/lax/stack.py index 509ffeff9d8c..ab7fb735b6ee 100644 --- a/jax/_src/lax/stack.py +++ b/jax/_src/lax/stack.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/svd.py b/jax/_src/lax/svd.py index 0cd9f04a5756..ae78bc2df72b 100644 --- a/jax/_src/lax/svd.py +++ b/jax/_src/lax/svd.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/utils.py b/jax/_src/lax/utils.py index 5408008fe9cb..b081eecf63ff 100644 --- a/jax/_src/lax/utils.py +++ b/jax/_src/lax/utils.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax/windowed_reductions.py b/jax/_src/lax/windowed_reductions.py index ad8e6d64b280..0e662538fc6a 100644 --- a/jax/_src/lax/windowed_reductions.py +++ b/jax/_src/lax/windowed_reductions.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lax_reference.py b/jax/_src/lax_reference.py index 30b347181e85..27488194d98a 100644 --- a/jax/_src/lax_reference.py +++ b/jax/_src/lax_reference.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lib/__init__.py b/jax/_src/lib/__init__.py index ea68b3abae03..06dbc964d432 100644 --- a/jax/_src/lib/__init__.py +++ b/jax/_src/lib/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lib/mlir/__init__.py b/jax/_src/lib/mlir/__init__.py index 7d1babd1efa9..e3f6627381f5 100644 --- a/jax/_src/lib/mlir/__init__.py +++ b/jax/_src/lib/mlir/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lib/mlir/dialects/__init__.py b/jax/_src/lib/mlir/dialects/__init__.py index 81d892a24d58..aa9348d3c91d 100644 --- a/jax/_src/lib/mlir/dialects/__init__.py +++ b/jax/_src/lib/mlir/dialects/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/lib/xla_bridge.py b/jax/_src/lib/xla_bridge.py index 38e648296550..297f1e9d4ec6 100644 --- a/jax/_src/lib/xla_bridge.py +++ b/jax/_src/lib/xla_bridge.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/nn/__init__.py b/jax/_src/nn/__init__.py index 30257881ab32..60afaf5a8da1 100644 --- a/jax/_src/nn/__init__.py +++ b/jax/_src/nn/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/nn/functions.py b/jax/_src/nn/functions.py index cd7e3ba7a238..e88a1e23056f 100644 --- a/jax/_src/nn/functions.py +++ b/jax/_src/nn/functions.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/nn/initializers.py b/jax/_src/nn/initializers.py index f46703edb812..c3330ac41979 100644 --- a/jax/_src/nn/initializers.py +++ b/jax/_src/nn/initializers.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/numpy/__init__.py b/jax/_src/numpy/__init__.py index b0c7da3d7725..0a0a8260408e 100644 --- a/jax/_src/numpy/__init__.py +++ b/jax/_src/numpy/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/numpy/fft.py b/jax/_src/numpy/fft.py index f36e9de15bf8..e2a31f1684c1 100644 --- a/jax/_src/numpy/fft.py +++ b/jax/_src/numpy/fft.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/numpy/index_tricks.py b/jax/_src/numpy/index_tricks.py index 379224d88225..ee0a1d0e1810 100644 --- a/jax/_src/numpy/index_tricks.py +++ b/jax/_src/numpy/index_tricks.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/numpy/lax_numpy.py b/jax/_src/numpy/lax_numpy.py index d3fb050eaf57..c35e762b7e24 100644 --- a/jax/_src/numpy/lax_numpy.py +++ b/jax/_src/numpy/lax_numpy.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/numpy/linalg.py b/jax/_src/numpy/linalg.py index 97827576e8d0..de063d35b9de 100644 --- a/jax/_src/numpy/linalg.py +++ b/jax/_src/numpy/linalg.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/numpy/ndarray.py b/jax/_src/numpy/ndarray.py index caef0ced3ed2..a4c6926aea0d 100644 --- a/jax/_src/numpy/ndarray.py +++ b/jax/_src/numpy/ndarray.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/numpy/polynomial.py b/jax/_src/numpy/polynomial.py index d1aca3c202d0..ed483a889998 100644 --- a/jax/_src/numpy/polynomial.py +++ b/jax/_src/numpy/polynomial.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/numpy/reductions.py b/jax/_src/numpy/reductions.py index 48e55a076182..00834edfd34f 100644 --- a/jax/_src/numpy/reductions.py +++ b/jax/_src/numpy/reductions.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/numpy/setops.py b/jax/_src/numpy/setops.py index b92866808649..9d774d196c8e 100644 --- a/jax/_src/numpy/setops.py +++ b/jax/_src/numpy/setops.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/numpy/ufuncs.py b/jax/_src/numpy/ufuncs.py index 67d5290cb724..ae68e3613298 100644 --- a/jax/_src/numpy/ufuncs.py +++ b/jax/_src/numpy/ufuncs.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/numpy/util.py b/jax/_src/numpy/util.py index 3f4c970a0075..0d1d3ac1c2fe 100644 --- a/jax/_src/numpy/util.py +++ b/jax/_src/numpy/util.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/numpy/vectorize.py b/jax/_src/numpy/vectorize.py index 765b9d109141..88486c655322 100644 --- a/jax/_src/numpy/vectorize.py +++ b/jax/_src/numpy/vectorize.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/ops/__init__.py b/jax/_src/ops/__init__.py index 30257881ab32..60afaf5a8da1 100644 --- a/jax/_src/ops/__init__.py +++ b/jax/_src/ops/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/ops/scatter.py b/jax/_src/ops/scatter.py index 83f251fcc127..5a2381ddcf30 100644 --- a/jax/_src/ops/scatter.py +++ b/jax/_src/ops/scatter.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/pretty_printer.py b/jax/_src/pretty_printer.py index 9a76e30a4dd8..479484821957 100644 --- a/jax/_src/pretty_printer.py +++ b/jax/_src/pretty_printer.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/prng.py b/jax/_src/prng.py index b2d89ef88d98..136c6b9e66a3 100644 --- a/jax/_src/prng.py +++ b/jax/_src/prng.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/profiler.py b/jax/_src/profiler.py index 7cf84f7bd313..05aaa8dd05d6 100644 --- a/jax/_src/profiler.py +++ b/jax/_src/profiler.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/public_test_util.py b/jax/_src/public_test_util.py index 2fa0cbd2f747..0213917f8784 100644 --- a/jax/_src/public_test_util.py +++ b/jax/_src/public_test_util.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/random.py b/jax/_src/random.py index fa2829b22464..494c3a0a7939 100644 --- a/jax/_src/random.py +++ b/jax/_src/random.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/__init__.py b/jax/_src/scipy/__init__.py index 30257881ab32..60afaf5a8da1 100644 --- a/jax/_src/scipy/__init__.py +++ b/jax/_src/scipy/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/cluster/__init__.py b/jax/_src/scipy/cluster/__init__.py index bb545f476ea3..37ee0c309fae 100644 --- a/jax/_src/scipy/cluster/__init__.py +++ b/jax/_src/scipy/cluster/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/cluster/vq.py b/jax/_src/scipy/cluster/vq.py index 94de1e6673cc..ec57a10af56f 100644 --- a/jax/_src/scipy/cluster/vq.py +++ b/jax/_src/scipy/cluster/vq.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/fft.py b/jax/_src/scipy/fft.py index 459e3e6f1b77..50893aff4d77 100644 --- a/jax/_src/scipy/fft.py +++ b/jax/_src/scipy/fft.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/interpolate/__init__.py b/jax/_src/scipy/interpolate/__init__.py index bb545f476ea3..37ee0c309fae 100644 --- a/jax/_src/scipy/interpolate/__init__.py +++ b/jax/_src/scipy/interpolate/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/linalg.py b/jax/_src/scipy/linalg.py index 44eaf51e81cc..2801325fcd83 100644 --- a/jax/_src/scipy/linalg.py +++ b/jax/_src/scipy/linalg.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/ndimage.py b/jax/_src/scipy/ndimage.py index c5f91aae284d..9e66c78eae99 100644 --- a/jax/_src/scipy/ndimage.py +++ b/jax/_src/scipy/ndimage.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/optimize/__init__.py b/jax/_src/scipy/optimize/__init__.py index 30257881ab32..60afaf5a8da1 100644 --- a/jax/_src/scipy/optimize/__init__.py +++ b/jax/_src/scipy/optimize/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/optimize/_lbfgs.py b/jax/_src/scipy/optimize/_lbfgs.py index 61b8cb51da55..ffe30362c021 100644 --- a/jax/_src/scipy/optimize/_lbfgs.py +++ b/jax/_src/scipy/optimize/_lbfgs.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/optimize/bfgs.py b/jax/_src/scipy/optimize/bfgs.py index 197e3844cb9a..224361d6d00f 100644 --- a/jax/_src/scipy/optimize/bfgs.py +++ b/jax/_src/scipy/optimize/bfgs.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/optimize/line_search.py b/jax/_src/scipy/optimize/line_search.py index 0760b5f89032..a139d0d75713 100644 --- a/jax/_src/scipy/optimize/line_search.py +++ b/jax/_src/scipy/optimize/line_search.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/optimize/minimize.py b/jax/_src/scipy/optimize/minimize.py index 09d8a5f3790a..e0d4f043501d 100644 --- a/jax/_src/scipy/optimize/minimize.py +++ b/jax/_src/scipy/optimize/minimize.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/signal.py b/jax/_src/scipy/signal.py index ebf58be9ae1d..c276a813f51b 100644 --- a/jax/_src/scipy/signal.py +++ b/jax/_src/scipy/signal.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/sparse/__init__.py b/jax/_src/scipy/sparse/__init__.py index 30257881ab32..60afaf5a8da1 100644 --- a/jax/_src/scipy/sparse/__init__.py +++ b/jax/_src/scipy/sparse/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/sparse/linalg.py b/jax/_src/scipy/sparse/linalg.py index e9fddd05683c..1162b2f7a7ec 100644 --- a/jax/_src/scipy/sparse/linalg.py +++ b/jax/_src/scipy/sparse/linalg.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/special.py b/jax/_src/scipy/special.py index 6d1576e64a38..703376bc533f 100644 --- a/jax/_src/scipy/special.py +++ b/jax/_src/scipy/special.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/__init__.py b/jax/_src/scipy/stats/__init__.py index 30257881ab32..60afaf5a8da1 100644 --- a/jax/_src/scipy/stats/__init__.py +++ b/jax/_src/scipy/stats/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/bernoulli.py b/jax/_src/scipy/stats/bernoulli.py index 877c997817d7..ede4a6691ca4 100644 --- a/jax/_src/scipy/stats/bernoulli.py +++ b/jax/_src/scipy/stats/bernoulli.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/beta.py b/jax/_src/scipy/stats/beta.py index b62f108a4018..4fdcd3c636fb 100644 --- a/jax/_src/scipy/stats/beta.py +++ b/jax/_src/scipy/stats/beta.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/betabinom.py b/jax/_src/scipy/stats/betabinom.py index f2366ed7d768..b260fa808f1a 100644 --- a/jax/_src/scipy/stats/betabinom.py +++ b/jax/_src/scipy/stats/betabinom.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/cauchy.py b/jax/_src/scipy/stats/cauchy.py index 85179c858032..3c92c6e7ff53 100644 --- a/jax/_src/scipy/stats/cauchy.py +++ b/jax/_src/scipy/stats/cauchy.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/chi2.py b/jax/_src/scipy/stats/chi2.py index f74ffd43e281..c38e92158535 100644 --- a/jax/_src/scipy/stats/chi2.py +++ b/jax/_src/scipy/stats/chi2.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/dirichlet.py b/jax/_src/scipy/stats/dirichlet.py index cbdfece02458..269e8f2991bf 100644 --- a/jax/_src/scipy/stats/dirichlet.py +++ b/jax/_src/scipy/stats/dirichlet.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/expon.py b/jax/_src/scipy/stats/expon.py index bf4a7afb4c0d..785224c9be2f 100644 --- a/jax/_src/scipy/stats/expon.py +++ b/jax/_src/scipy/stats/expon.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/gamma.py b/jax/_src/scipy/stats/gamma.py index 52e96753ab07..f7b7ae9dbc8a 100644 --- a/jax/_src/scipy/stats/gamma.py +++ b/jax/_src/scipy/stats/gamma.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/gennorm.py b/jax/_src/scipy/stats/gennorm.py index d4b55b050f94..95744fc3441a 100644 --- a/jax/_src/scipy/stats/gennorm.py +++ b/jax/_src/scipy/stats/gennorm.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/geom.py b/jax/_src/scipy/stats/geom.py index b51c381fc172..126acfe6f172 100644 --- a/jax/_src/scipy/stats/geom.py +++ b/jax/_src/scipy/stats/geom.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/kde.py b/jax/_src/scipy/stats/kde.py index e0bc8d9b7804..2276c4624403 100644 --- a/jax/_src/scipy/stats/kde.py +++ b/jax/_src/scipy/stats/kde.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/laplace.py b/jax/_src/scipy/stats/laplace.py index 827ffd3f2bb0..1c03f3a18876 100644 --- a/jax/_src/scipy/stats/laplace.py +++ b/jax/_src/scipy/stats/laplace.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/logistic.py b/jax/_src/scipy/stats/logistic.py index 5d81a2f6d94f..6657d0f800ed 100644 --- a/jax/_src/scipy/stats/logistic.py +++ b/jax/_src/scipy/stats/logistic.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/multinomial.py b/jax/_src/scipy/stats/multinomial.py index 4d468690e220..3ad75f962c2d 100644 --- a/jax/_src/scipy/stats/multinomial.py +++ b/jax/_src/scipy/stats/multinomial.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/multivariate_normal.py b/jax/_src/scipy/stats/multivariate_normal.py index ffcaa00ad126..99e709a25cfe 100644 --- a/jax/_src/scipy/stats/multivariate_normal.py +++ b/jax/_src/scipy/stats/multivariate_normal.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/nbinom.py b/jax/_src/scipy/stats/nbinom.py index 8b4da9dc7dcc..48c19344edbb 100644 --- a/jax/_src/scipy/stats/nbinom.py +++ b/jax/_src/scipy/stats/nbinom.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/norm.py b/jax/_src/scipy/stats/norm.py index 952adc082370..1ead3aaf7654 100644 --- a/jax/_src/scipy/stats/norm.py +++ b/jax/_src/scipy/stats/norm.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/pareto.py b/jax/_src/scipy/stats/pareto.py index 6e86049816bb..c258446cf5a7 100644 --- a/jax/_src/scipy/stats/pareto.py +++ b/jax/_src/scipy/stats/pareto.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/poisson.py b/jax/_src/scipy/stats/poisson.py index c509ab5210b4..bdc0e8cd4230 100644 --- a/jax/_src/scipy/stats/poisson.py +++ b/jax/_src/scipy/stats/poisson.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/t.py b/jax/_src/scipy/stats/t.py index 4056ebe78144..b9484acb2ebd 100644 --- a/jax/_src/scipy/stats/t.py +++ b/jax/_src/scipy/stats/t.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/scipy/stats/uniform.py b/jax/_src/scipy/stats/uniform.py index 47cb1e1d9164..df315326b223 100644 --- a/jax/_src/scipy/stats/uniform.py +++ b/jax/_src/scipy/stats/uniform.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/source_info_util.py b/jax/_src/source_info_util.py index d59df543dc59..799a1cb7023d 100644 --- a/jax/_src/source_info_util.py +++ b/jax/_src/source_info_util.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/stages.py b/jax/_src/stages.py index 94d9feb1eca2..a6d6e1981aba 100644 --- a/jax/_src/stages.py +++ b/jax/_src/stages.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/state/__init__.py b/jax/_src/state/__init__.py index 8e48c357e02c..55098306f9d6 100644 --- a/jax/_src/state/__init__.py +++ b/jax/_src/state/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/state/discharge.py b/jax/_src/state/discharge.py index 98edd0a7df41..96725a1e2eac 100644 --- a/jax/_src/state/discharge.py +++ b/jax/_src/state/discharge.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/state/primitives.py b/jax/_src/state/primitives.py index ea1b1034d895..e4dd42d49efd 100644 --- a/jax/_src/state/primitives.py +++ b/jax/_src/state/primitives.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/state/types.py b/jax/_src/state/types.py index d11f7117ea5d..e16d11fcd067 100644 --- a/jax/_src/state/types.py +++ b/jax/_src/state/types.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/test_util.py b/jax/_src/test_util.py index 91f89f321aa6..686390886f10 100644 --- a/jax/_src/test_util.py +++ b/jax/_src/test_util.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/traceback_util.py b/jax/_src/traceback_util.py index 44c6a347adf1..51c09e12c7f0 100644 --- a/jax/_src/traceback_util.py +++ b/jax/_src/traceback_util.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/tree_util.py b/jax/_src/tree_util.py index 3d7e259df34a..6ef91ff8688f 100644 --- a/jax/_src/tree_util.py +++ b/jax/_src/tree_util.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/typing.py b/jax/_src/typing.py index 16ec4f7c47a7..4cb94ff7aaba 100644 --- a/jax/_src/typing.py +++ b/jax/_src/typing.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/_src/util.py b/jax/_src/util.py index 900208b718ea..926a3584b0a3 100644 --- a/jax/_src/util.py +++ b/jax/_src/util.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/abstract_arrays.py b/jax/abstract_arrays.py index de8ddf7a2643..a9072fb6178c 100644 --- a/jax/abstract_arrays.py +++ b/jax/abstract_arrays.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/ad_checkpoint.py b/jax/ad_checkpoint.py index 18da9c4bdb54..b7413954ab8c 100644 --- a/jax/ad_checkpoint.py +++ b/jax/ad_checkpoint.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/api_util.py b/jax/api_util.py index 2fe9b0025163..77b3ae36f90b 100644 --- a/jax/api_util.py +++ b/jax/api_util.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/cloud_tpu_init.py b/jax/cloud_tpu_init.py index 0a4337cd8088..8cc49ac3b73f 100644 --- a/jax/cloud_tpu_init.py +++ b/jax/cloud_tpu_init.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/collect_profile.py b/jax/collect_profile.py index 452185deafe1..c61d1a347ace 100644 --- a/jax/collect_profile.py +++ b/jax/collect_profile.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/config.py b/jax/config.py index 8486017b6799..b8283200c942 100644 --- a/jax/config.py +++ b/jax/config.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/core.py b/jax/core.py index e585607f1e16..092c39938f89 100644 --- a/jax/core.py +++ b/jax/core.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/custom_batching.py b/jax/custom_batching.py index 61cdd71a3b51..a4850f04c2ec 100644 --- a/jax/custom_batching.py +++ b/jax/custom_batching.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/custom_derivatives.py b/jax/custom_derivatives.py index 51f55e0ef591..a178d7d4b961 100644 --- a/jax/custom_derivatives.py +++ b/jax/custom_derivatives.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/custom_transpose.py b/jax/custom_transpose.py index 76a75a771804..311139da2567 100644 --- a/jax/custom_transpose.py +++ b/jax/custom_transpose.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/debug.py b/jax/debug.py index bf328a8c8467..bd428bdbc864 100644 --- a/jax/debug.py +++ b/jax/debug.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/distributed.py b/jax/distributed.py index 2d4720c568fa..284ae6f95f48 100644 --- a/jax/distributed.py +++ b/jax/distributed.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/dlpack.py b/jax/dlpack.py index 50d03abfad81..707e966ee243 100644 --- a/jax/dlpack.py +++ b/jax/dlpack.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/dtypes.py b/jax/dtypes.py index cc7589c77b7e..dc321bdb4bd1 100644 --- a/jax/dtypes.py +++ b/jax/dtypes.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/errors.py b/jax/errors.py index b8afb8e87feb..80775a0c16fe 100644 --- a/jax/errors.py +++ b/jax/errors.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/example_libraries/__init__.py b/jax/example_libraries/__init__.py index 4fbd93bb2ca4..577c9dcb20e0 100644 --- a/jax/example_libraries/__init__.py +++ b/jax/example_libraries/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/example_libraries/optimizers.py b/jax/example_libraries/optimizers.py index 239b18a14bba..46eb7e8a6e3b 100644 --- a/jax/example_libraries/optimizers.py +++ b/jax/example_libraries/optimizers.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/example_libraries/stax.py b/jax/example_libraries/stax.py index ba711b886d46..e5bf38c6a69f 100644 --- a/jax/example_libraries/stax.py +++ b/jax/example_libraries/stax.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/__init__.py b/jax/experimental/__init__.py index a1510dcbbf08..7d55b9de0661 100644 --- a/jax/experimental/__init__.py +++ b/jax/experimental/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/array.py b/jax/experimental/array.py index b2b2aed16dc7..1f325cda3f5f 100644 --- a/jax/experimental/array.py +++ b/jax/experimental/array.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/callback.py b/jax/experimental/callback.py index 33da94adc91c..ccbc9f348863 100644 --- a/jax/experimental/callback.py +++ b/jax/experimental/callback.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/checkify.py b/jax/experimental/checkify.py index 9987d0f106e3..c72e032ec447 100644 --- a/jax/experimental/checkify.py +++ b/jax/experimental/checkify.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/compilation_cache/__init__.py b/jax/experimental/compilation_cache/__init__.py index 4fbd93bb2ca4..577c9dcb20e0 100644 --- a/jax/experimental/compilation_cache/__init__.py +++ b/jax/experimental/compilation_cache/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/compilation_cache/cache_interface.py b/jax/experimental/compilation_cache/cache_interface.py index 5359249bf1be..81e2fd5a640a 100644 --- a/jax/experimental/compilation_cache/cache_interface.py +++ b/jax/experimental/compilation_cache/cache_interface.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/compilation_cache/compilation_cache.py b/jax/experimental/compilation_cache/compilation_cache.py index 84dda9f3c0f7..cd3653bc9ea7 100644 --- a/jax/experimental/compilation_cache/compilation_cache.py +++ b/jax/experimental/compilation_cache/compilation_cache.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/compilation_cache/gfile_cache.py b/jax/experimental/compilation_cache/gfile_cache.py index 9c3dc4579fe8..03dcabfdbb82 100644 --- a/jax/experimental/compilation_cache/gfile_cache.py +++ b/jax/experimental/compilation_cache/gfile_cache.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/gda_serialization/__init__.py b/jax/experimental/gda_serialization/__init__.py index 4fbd93bb2ca4..577c9dcb20e0 100644 --- a/jax/experimental/gda_serialization/__init__.py +++ b/jax/experimental/gda_serialization/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/gda_serialization/serialization.py b/jax/experimental/gda_serialization/serialization.py index 1797a2d06678..be0ae2b888f4 100644 --- a/jax/experimental/gda_serialization/serialization.py +++ b/jax/experimental/gda_serialization/serialization.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/gda_serialization/serialization_test.py b/jax/experimental/gda_serialization/serialization_test.py index 1242521ae048..f3bfd2143779 100644 --- a/jax/experimental/gda_serialization/serialization_test.py +++ b/jax/experimental/gda_serialization/serialization_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/global_device_array.py b/jax/experimental/global_device_array.py index 4a52f2d1ec22..1993f450a2d0 100644 --- a/jax/experimental/global_device_array.py +++ b/jax/experimental/global_device_array.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/host_callback.py b/jax/experimental/host_callback.py index b55e0bfc85c0..2ba1f4c8f93d 100644 --- a/jax/experimental/host_callback.py +++ b/jax/experimental/host_callback.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/BUILD b/jax/experimental/jax2tf/BUILD index 308a66c67850..90315f7ca02c 100644 --- a/jax/experimental/jax2tf/BUILD +++ b/jax/experimental/jax2tf/BUILD @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/__init__.py b/jax/experimental/jax2tf/__init__.py index c83eda7dec27..6a2dd2fd71b1 100644 --- a/jax/experimental/jax2tf/__init__.py +++ b/jax/experimental/jax2tf/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/call_tf.py b/jax/experimental/jax2tf/call_tf.py index 1d7b4f267f64..e9ea8af012f0 100644 --- a/jax/experimental/jax2tf/call_tf.py +++ b/jax/experimental/jax2tf/call_tf.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/examples/__init__.py b/jax/experimental/jax2tf/examples/__init__.py index 4fbd93bb2ca4..577c9dcb20e0 100644 --- a/jax/experimental/jax2tf/examples/__init__.py +++ b/jax/experimental/jax2tf/examples/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/examples/keras_reuse_main.py b/jax/experimental/jax2tf/examples/keras_reuse_main.py index aae32716b0f9..52c3203c6363 100644 --- a/jax/experimental/jax2tf/examples/keras_reuse_main.py +++ b/jax/experimental/jax2tf/examples/keras_reuse_main.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/examples/keras_reuse_main_test.py b/jax/experimental/jax2tf/examples/keras_reuse_main_test.py index 540821dc442b..7e282f0f25f7 100644 --- a/jax/experimental/jax2tf/examples/keras_reuse_main_test.py +++ b/jax/experimental/jax2tf/examples/keras_reuse_main_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/examples/mnist_lib.py b/jax/experimental/jax2tf/examples/mnist_lib.py index cbc5bbe563a8..3e7f6ea1f000 100644 --- a/jax/experimental/jax2tf/examples/mnist_lib.py +++ b/jax/experimental/jax2tf/examples/mnist_lib.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/examples/saved_model_lib.py b/jax/experimental/jax2tf/examples/saved_model_lib.py index 987cc28d316d..63eb6faa2528 100644 --- a/jax/experimental/jax2tf/examples/saved_model_lib.py +++ b/jax/experimental/jax2tf/examples/saved_model_lib.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/examples/saved_model_main.py b/jax/experimental/jax2tf/examples/saved_model_main.py index a2eeeb7eaab6..ed7b8eefd413 100644 --- a/jax/experimental/jax2tf/examples/saved_model_main.py +++ b/jax/experimental/jax2tf/examples/saved_model_main.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/examples/saved_model_main_test.py b/jax/experimental/jax2tf/examples/saved_model_main_test.py index e160e1ad2ad9..de7d21652fad 100644 --- a/jax/experimental/jax2tf/examples/saved_model_main_test.py +++ b/jax/experimental/jax2tf/examples/saved_model_main_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/examples/serving/__init__.py b/jax/experimental/jax2tf/examples/serving/__init__.py index 4fbd93bb2ca4..577c9dcb20e0 100644 --- a/jax/experimental/jax2tf/examples/serving/__init__.py +++ b/jax/experimental/jax2tf/examples/serving/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/examples/serving/model_server_request.py b/jax/experimental/jax2tf/examples/serving/model_server_request.py index 63c29a96ce2d..8fd0b6fc66d2 100644 --- a/jax/experimental/jax2tf/examples/serving/model_server_request.py +++ b/jax/experimental/jax2tf/examples/serving/model_server_request.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/examples/tf_js/quickdraw/input_pipeline.py b/jax/experimental/jax2tf/examples/tf_js/quickdraw/input_pipeline.py index a18e474d6b19..19c564bdac70 100644 --- a/jax/experimental/jax2tf/examples/tf_js/quickdraw/input_pipeline.py +++ b/jax/experimental/jax2tf/examples/tf_js/quickdraw/input_pipeline.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/examples/tf_js/quickdraw/quickdraw.py b/jax/experimental/jax2tf/examples/tf_js/quickdraw/quickdraw.py index 25e30eb30cd4..151610728eb2 100644 --- a/jax/experimental/jax2tf/examples/tf_js/quickdraw/quickdraw.py +++ b/jax/experimental/jax2tf/examples/tf_js/quickdraw/quickdraw.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/examples/tflite/mnist/mnist.py b/jax/experimental/jax2tf/examples/tflite/mnist/mnist.py index 0cc346e7c8fc..5452ea13dd02 100644 --- a/jax/experimental/jax2tf/examples/tflite/mnist/mnist.py +++ b/jax/experimental/jax2tf/examples/tflite/mnist/mnist.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/impl_no_xla.py b/jax/experimental/jax2tf/impl_no_xla.py index 3fb3cd3686f8..5adfed582400 100644 --- a/jax/experimental/jax2tf/impl_no_xla.py +++ b/jax/experimental/jax2tf/impl_no_xla.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/jax2tf.py b/jax/experimental/jax2tf/jax2tf.py index 4021a5492e43..19043300026e 100644 --- a/jax/experimental/jax2tf/jax2tf.py +++ b/jax/experimental/jax2tf/jax2tf.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/shape_poly.py b/jax/experimental/jax2tf/shape_poly.py index bbd1e0c38be1..6e5834709482 100644 --- a/jax/experimental/jax2tf/shape_poly.py +++ b/jax/experimental/jax2tf/shape_poly.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/__init__.py b/jax/experimental/jax2tf/tests/__init__.py index 30257881ab32..60afaf5a8da1 100644 --- a/jax/experimental/jax2tf/tests/__init__.py +++ b/jax/experimental/jax2tf/tests/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/call_tf_test.py b/jax/experimental/jax2tf/tests/call_tf_test.py index e88c4c81e2af..65771e595c74 100644 --- a/jax/experimental/jax2tf/tests/call_tf_test.py +++ b/jax/experimental/jax2tf/tests/call_tf_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/control_flow_ops_test.py b/jax/experimental/jax2tf/tests/control_flow_ops_test.py index 6d87a31faa45..72ce456d09db 100644 --- a/jax/experimental/jax2tf/tests/control_flow_ops_test.py +++ b/jax/experimental/jax2tf/tests/control_flow_ops_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/converters.py b/jax/experimental/jax2tf/tests/converters.py index 10dd7346a430..61fcedea773e 100644 --- a/jax/experimental/jax2tf/tests/converters.py +++ b/jax/experimental/jax2tf/tests/converters.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/flax_models/BUILD b/jax/experimental/jax2tf/tests/flax_models/BUILD index cda8184e652a..761c27154620 100644 --- a/jax/experimental/jax2tf/tests/flax_models/BUILD +++ b/jax/experimental/jax2tf/tests/flax_models/BUILD @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/flax_models/actor_critic.py b/jax/experimental/jax2tf/tests/flax_models/actor_critic.py index 57533bf29b74..b2e537bc6f37 100644 --- a/jax/experimental/jax2tf/tests/flax_models/actor_critic.py +++ b/jax/experimental/jax2tf/tests/flax_models/actor_critic.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/flax_models/bilstm_classifier.py b/jax/experimental/jax2tf/tests/flax_models/bilstm_classifier.py index c023c8bb29ce..022f221cd29a 100644 --- a/jax/experimental/jax2tf/tests/flax_models/bilstm_classifier.py +++ b/jax/experimental/jax2tf/tests/flax_models/bilstm_classifier.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/flax_models/cnn.py b/jax/experimental/jax2tf/tests/flax_models/cnn.py index 8e9c82aae063..dea8f23a2194 100644 --- a/jax/experimental/jax2tf/tests/flax_models/cnn.py +++ b/jax/experimental/jax2tf/tests/flax_models/cnn.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/flax_models/gnn.py b/jax/experimental/jax2tf/tests/flax_models/gnn.py index 7bca5cc2c2e3..67cad2c62bfe 100644 --- a/jax/experimental/jax2tf/tests/flax_models/gnn.py +++ b/jax/experimental/jax2tf/tests/flax_models/gnn.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/flax_models/resnet.py b/jax/experimental/jax2tf/tests/flax_models/resnet.py index 84243e076721..b03b66f659d2 100644 --- a/jax/experimental/jax2tf/tests/flax_models/resnet.py +++ b/jax/experimental/jax2tf/tests/flax_models/resnet.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/flax_models/seq2seq_lstm.py b/jax/experimental/jax2tf/tests/flax_models/seq2seq_lstm.py index 91fce99fa16e..633240c8af06 100644 --- a/jax/experimental/jax2tf/tests/flax_models/seq2seq_lstm.py +++ b/jax/experimental/jax2tf/tests/flax_models/seq2seq_lstm.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/flax_models/transformer_lm1b.py b/jax/experimental/jax2tf/tests/flax_models/transformer_lm1b.py index 282ee4afe541..4ccdd3ab9e0f 100644 --- a/jax/experimental/jax2tf/tests/flax_models/transformer_lm1b.py +++ b/jax/experimental/jax2tf/tests/flax_models/transformer_lm1b.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/flax_models/transformer_nlp_seq.py b/jax/experimental/jax2tf/tests/flax_models/transformer_nlp_seq.py index f2ff61a158b5..5fd9a0d40dc5 100644 --- a/jax/experimental/jax2tf/tests/flax_models/transformer_nlp_seq.py +++ b/jax/experimental/jax2tf/tests/flax_models/transformer_nlp_seq.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/flax_models/transformer_wmt.py b/jax/experimental/jax2tf/tests/flax_models/transformer_wmt.py index 9111b0fb8602..b620bade14b4 100644 --- a/jax/experimental/jax2tf/tests/flax_models/transformer_wmt.py +++ b/jax/experimental/jax2tf/tests/flax_models/transformer_wmt.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/flax_models/vae.py b/jax/experimental/jax2tf/tests/flax_models/vae.py index 65cbebec79cb..2a43035e4ff5 100644 --- a/jax/experimental/jax2tf/tests/flax_models/vae.py +++ b/jax/experimental/jax2tf/tests/flax_models/vae.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/jax2tf_limitations.py b/jax/experimental/jax2tf/tests/jax2tf_limitations.py index 1fa62480a2db..51f0a816bd74 100644 --- a/jax/experimental/jax2tf/tests/jax2tf_limitations.py +++ b/jax/experimental/jax2tf/tests/jax2tf_limitations.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/jax2tf_test.py b/jax/experimental/jax2tf/tests/jax2tf_test.py index e21ac392cb5b..7b8fa4e12138 100644 --- a/jax/experimental/jax2tf/tests/jax2tf_test.py +++ b/jax/experimental/jax2tf/tests/jax2tf_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py b/jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py index 29c99491ed94..d6d8663a521a 100644 --- a/jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py +++ b/jax/experimental/jax2tf/tests/jax_primitives_coverage_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/model_harness.py b/jax/experimental/jax2tf/tests/model_harness.py index cfdd079aefdf..65e9225f9c08 100644 --- a/jax/experimental/jax2tf/tests/model_harness.py +++ b/jax/experimental/jax2tf/tests/model_harness.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/models_test_main.py b/jax/experimental/jax2tf/tests/models_test_main.py index efe7173cbf28..9a67ae35b0fc 100644 --- a/jax/experimental/jax2tf/tests/models_test_main.py +++ b/jax/experimental/jax2tf/tests/models_test_main.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/primitive_harness.py b/jax/experimental/jax2tf/tests/primitive_harness.py index ea70485a57bd..868cb5b57579 100644 --- a/jax/experimental/jax2tf/tests/primitive_harness.py +++ b/jax/experimental/jax2tf/tests/primitive_harness.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/primitives_test.py b/jax/experimental/jax2tf/tests/primitives_test.py index c2a57999d307..20d1e01ef793 100644 --- a/jax/experimental/jax2tf/tests/primitives_test.py +++ b/jax/experimental/jax2tf/tests/primitives_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/savedmodel_test.py b/jax/experimental/jax2tf/tests/savedmodel_test.py index 4f55caa4fd6a..ab5039f4fc41 100644 --- a/jax/experimental/jax2tf/tests/savedmodel_test.py +++ b/jax/experimental/jax2tf/tests/savedmodel_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/shape_poly_test.py b/jax/experimental/jax2tf/tests/shape_poly_test.py index ff74f774e18b..6e90b00ff458 100644 --- a/jax/experimental/jax2tf/tests/shape_poly_test.py +++ b/jax/experimental/jax2tf/tests/shape_poly_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/sharding_test.py b/jax/experimental/jax2tf/tests/sharding_test.py index 246da8ffd193..71e7eef8f0b7 100644 --- a/jax/experimental/jax2tf/tests/sharding_test.py +++ b/jax/experimental/jax2tf/tests/sharding_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jax2tf/tests/tf_test_util.py b/jax/experimental/jax2tf/tests/tf_test_util.py index 990d416583fc..53ad17ae5d7a 100644 --- a/jax/experimental/jax2tf/tests/tf_test_util.py +++ b/jax/experimental/jax2tf/tests/tf_test_util.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/jet.py b/jax/experimental/jet.py index f3f36f0315b3..886413ddc75e 100644 --- a/jax/experimental/jet.py +++ b/jax/experimental/jet.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/maps.py b/jax/experimental/maps.py index 8b13fc9c1e28..22b1ae2770d3 100644 --- a/jax/experimental/maps.py +++ b/jax/experimental/maps.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/multihost_utils.py b/jax/experimental/multihost_utils.py index 75665a0832c4..74e791b7e2ea 100644 --- a/jax/experimental/multihost_utils.py +++ b/jax/experimental/multihost_utils.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/ode.py b/jax/experimental/ode.py index c9e029029236..ff73236449cc 100644 --- a/jax/experimental/ode.py +++ b/jax/experimental/ode.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/pjit.py b/jax/experimental/pjit.py index 34a7a6a3605a..dc253974f35a 100644 --- a/jax/experimental/pjit.py +++ b/jax/experimental/pjit.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/sharding.py b/jax/experimental/sharding.py index 88c2d96e57bc..dafb4b3e086b 100644 --- a/jax/experimental/sharding.py +++ b/jax/experimental/sharding.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/sparse/__init__.py b/jax/experimental/sparse/__init__.py index 4bfaafc8d6e1..1887c4cc0dbc 100644 --- a/jax/experimental/sparse/__init__.py +++ b/jax/experimental/sparse/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/sparse/_base.py b/jax/experimental/sparse/_base.py index 1171f2a4b187..a650cd4fb1df 100644 --- a/jax/experimental/sparse/_base.py +++ b/jax/experimental/sparse/_base.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/sparse/ad.py b/jax/experimental/sparse/ad.py index 10499d0dfe80..6adaee8f9f4b 100644 --- a/jax/experimental/sparse/ad.py +++ b/jax/experimental/sparse/ad.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/sparse/api.py b/jax/experimental/sparse/api.py index b312a413cad3..49551fadcf8d 100644 --- a/jax/experimental/sparse/api.py +++ b/jax/experimental/sparse/api.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/sparse/bcoo.py b/jax/experimental/sparse/bcoo.py index 321ee746e11b..82fb0f89a51c 100644 --- a/jax/experimental/sparse/bcoo.py +++ b/jax/experimental/sparse/bcoo.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/sparse/coo.py b/jax/experimental/sparse/coo.py index eaeee8f3ee4b..b5f5cba3392f 100644 --- a/jax/experimental/sparse/coo.py +++ b/jax/experimental/sparse/coo.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/sparse/csr.py b/jax/experimental/sparse/csr.py index 26f0c68e6626..1d536cca4b37 100644 --- a/jax/experimental/sparse/csr.py +++ b/jax/experimental/sparse/csr.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/sparse/linalg.py b/jax/experimental/sparse/linalg.py index de08f56c3ec2..da7f6e038d13 100644 --- a/jax/experimental/sparse/linalg.py +++ b/jax/experimental/sparse/linalg.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/sparse/random.py b/jax/experimental/sparse/random.py index d9bf532c3599..8b05e82c927b 100644 --- a/jax/experimental/sparse/random.py +++ b/jax/experimental/sparse/random.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/sparse/transform.py b/jax/experimental/sparse/transform.py index 43ada03f6286..11da9cfda852 100644 --- a/jax/experimental/sparse/transform.py +++ b/jax/experimental/sparse/transform.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/sparse/util.py b/jax/experimental/sparse/util.py index bf8434bbb7b5..5343f3333444 100644 --- a/jax/experimental/sparse/util.py +++ b/jax/experimental/sparse/util.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/experimental/x64_context.py b/jax/experimental/x64_context.py index d011a966755b..628eb7b6060b 100644 --- a/jax/experimental/x64_context.py +++ b/jax/experimental/x64_context.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/flatten_util.py b/jax/flatten_util.py index 297f218db3a3..0736659b7aa7 100644 --- a/jax/flatten_util.py +++ b/jax/flatten_util.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/image/__init__.py b/jax/image/__init__.py index 6911f2b7d4da..672de5f1fe0e 100644 --- a/jax/image/__init__.py +++ b/jax/image/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/interpreters/__init__.py b/jax/interpreters/__init__.py index 26d45a07f8ea..4fbd409374e8 100644 --- a/jax/interpreters/__init__.py +++ b/jax/interpreters/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/interpreters/ad.py b/jax/interpreters/ad.py index 93c2b2975c96..e50e9f151be3 100644 --- a/jax/interpreters/ad.py +++ b/jax/interpreters/ad.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/interpreters/batching.py b/jax/interpreters/batching.py index d08e6cd09470..f52a15271e7a 100644 --- a/jax/interpreters/batching.py +++ b/jax/interpreters/batching.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/interpreters/mlir.py b/jax/interpreters/mlir.py index 8502b3c0710a..1cf335689ba9 100644 --- a/jax/interpreters/mlir.py +++ b/jax/interpreters/mlir.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/interpreters/partial_eval.py b/jax/interpreters/partial_eval.py index cf4cd8a8bbed..8cc65f283293 100644 --- a/jax/interpreters/partial_eval.py +++ b/jax/interpreters/partial_eval.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/interpreters/pxla.py b/jax/interpreters/pxla.py index 6176d860f856..405cfe34ac5b 100644 --- a/jax/interpreters/pxla.py +++ b/jax/interpreters/pxla.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/interpreters/xla.py b/jax/interpreters/xla.py index bebb5205d284..f243a41d4980 100644 --- a/jax/interpreters/xla.py +++ b/jax/interpreters/xla.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/jaxpr_util.py b/jax/jaxpr_util.py index 532e1a51965e..be7591d3f2e2 100644 --- a/jax/jaxpr_util.py +++ b/jax/jaxpr_util.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/lax/__init__.py b/jax/lax/__init__.py index 53480a66f4ca..e83300141565 100644 --- a/jax/lax/__init__.py +++ b/jax/lax/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/lax/linalg.py b/jax/lax/linalg.py index 4bc61fd42469..28f4edb942e6 100644 --- a/jax/lax/linalg.py +++ b/jax/lax/linalg.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/lib/__init__.py b/jax/lib/__init__.py index 0ea575964a5d..9841d4cb26fd 100644 --- a/jax/lib/__init__.py +++ b/jax/lib/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/lib/xla_bridge.py b/jax/lib/xla_bridge.py index b1c042091771..e7f775e3d3f8 100644 --- a/jax/lib/xla_bridge.py +++ b/jax/lib/xla_bridge.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/linear_util.py b/jax/linear_util.py index d5691d2ace78..1d4599a38216 100644 --- a/jax/linear_util.py +++ b/jax/linear_util.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/nn/__init__.py b/jax/nn/__init__.py index 6fbc55c81e6a..73a10696779b 100644 --- a/jax/nn/__init__.py +++ b/jax/nn/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/nn/initializers.py b/jax/nn/initializers.py index 0ae9cd1a2f7f..cf2b50fce355 100644 --- a/jax/nn/initializers.py +++ b/jax/nn/initializers.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/numpy/__init__.py b/jax/numpy/__init__.py index e74c1165c6fd..531a0889ef32 100644 --- a/jax/numpy/__init__.py +++ b/jax/numpy/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/numpy/fft.py b/jax/numpy/fft.py index be33b038fc21..96574086dcfc 100644 --- a/jax/numpy/fft.py +++ b/jax/numpy/fft.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/numpy/linalg.py b/jax/numpy/linalg.py index 8fba2c9341d4..f0fe691a3c81 100644 --- a/jax/numpy/linalg.py +++ b/jax/numpy/linalg.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/ops/__init__.py b/jax/ops/__init__.py index cc181e52888b..38b8b21d4b75 100644 --- a/jax/ops/__init__.py +++ b/jax/ops/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/prng.py b/jax/prng.py index 7cc4d4a5c99d..1df004ce8dd2 100644 --- a/jax/prng.py +++ b/jax/prng.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/profiler.py b/jax/profiler.py index d81a778405e3..246ea90042d4 100644 --- a/jax/profiler.py +++ b/jax/profiler.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/random.py b/jax/random.py index 75ab9215b6c4..1b307dd7f0f9 100644 --- a/jax/random.py +++ b/jax/random.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/__init__.py b/jax/scipy/__init__.py index 0f504cf64b1f..47f5bc0cfad7 100644 --- a/jax/scipy/__init__.py +++ b/jax/scipy/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/cluster/__init__.py b/jax/scipy/cluster/__init__.py index c0c3ced15701..1ed987ad98c4 100644 --- a/jax/scipy/cluster/__init__.py +++ b/jax/scipy/cluster/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/cluster/vq.py b/jax/scipy/cluster/vq.py index 0b836b09c019..6966bd9198d5 100644 --- a/jax/scipy/cluster/vq.py +++ b/jax/scipy/cluster/vq.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/fft.py b/jax/scipy/fft.py index e88567bb60cc..759c780746b1 100644 --- a/jax/scipy/fft.py +++ b/jax/scipy/fft.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/interpolate/__init__.py b/jax/scipy/interpolate/__init__.py index 17253362a328..22cb3d77ece2 100644 --- a/jax/scipy/interpolate/__init__.py +++ b/jax/scipy/interpolate/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/linalg.py b/jax/scipy/linalg.py index 1a7b9cee5a10..dddf6a5b9579 100644 --- a/jax/scipy/linalg.py +++ b/jax/scipy/linalg.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/ndimage.py b/jax/scipy/ndimage.py index 3d822f08d866..ffc4970b8192 100644 --- a/jax/scipy/ndimage.py +++ b/jax/scipy/ndimage.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/optimize/__init__.py b/jax/scipy/optimize/__init__.py index 81e162497df0..b4ca718327ce 100644 --- a/jax/scipy/optimize/__init__.py +++ b/jax/scipy/optimize/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/signal.py b/jax/scipy/signal.py index 4fdec22d1601..34cdb7b25953 100644 --- a/jax/scipy/signal.py +++ b/jax/scipy/signal.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/sparse/__init__.py b/jax/scipy/sparse/__init__.py index 42170049fae0..a14fb441ea1e 100644 --- a/jax/scipy/sparse/__init__.py +++ b/jax/scipy/sparse/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/sparse/linalg.py b/jax/scipy/sparse/linalg.py index 3c1a0f390b7d..d982364978db 100644 --- a/jax/scipy/sparse/linalg.py +++ b/jax/scipy/sparse/linalg.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/special.py b/jax/scipy/special.py index 91694f9dfb91..9c2279ec101e 100644 --- a/jax/scipy/special.py +++ b/jax/scipy/special.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/__init__.py b/jax/scipy/stats/__init__.py index f9d749b20382..c57bcbb51d2e 100644 --- a/jax/scipy/stats/__init__.py +++ b/jax/scipy/stats/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/bernoulli.py b/jax/scipy/stats/bernoulli.py index 010835c81f19..0cbe5d603c2c 100644 --- a/jax/scipy/stats/bernoulli.py +++ b/jax/scipy/stats/bernoulli.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/beta.py b/jax/scipy/stats/beta.py index 1c4afecc7722..a8024a10ee1b 100644 --- a/jax/scipy/stats/beta.py +++ b/jax/scipy/stats/beta.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/betabinom.py b/jax/scipy/stats/betabinom.py index 75a1cc257bcf..ab122cad3b53 100644 --- a/jax/scipy/stats/betabinom.py +++ b/jax/scipy/stats/betabinom.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/cauchy.py b/jax/scipy/stats/cauchy.py index 4429d9bbcde2..8cdde458ab01 100644 --- a/jax/scipy/stats/cauchy.py +++ b/jax/scipy/stats/cauchy.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/chi2.py b/jax/scipy/stats/chi2.py index 58d26c87841d..99ca46ab52b1 100644 --- a/jax/scipy/stats/chi2.py +++ b/jax/scipy/stats/chi2.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/dirichlet.py b/jax/scipy/stats/dirichlet.py index 42153d4b7a70..86dbd0dc03a7 100644 --- a/jax/scipy/stats/dirichlet.py +++ b/jax/scipy/stats/dirichlet.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/expon.py b/jax/scipy/stats/expon.py index 550551445b56..19528bb0885f 100644 --- a/jax/scipy/stats/expon.py +++ b/jax/scipy/stats/expon.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/gamma.py b/jax/scipy/stats/gamma.py index 25962836bf6d..320e4e236be4 100644 --- a/jax/scipy/stats/gamma.py +++ b/jax/scipy/stats/gamma.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/gennorm.py b/jax/scipy/stats/gennorm.py index 2d069773de75..c35830a92eac 100644 --- a/jax/scipy/stats/gennorm.py +++ b/jax/scipy/stats/gennorm.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/geom.py b/jax/scipy/stats/geom.py index d69cbebc8220..92bb24626f54 100644 --- a/jax/scipy/stats/geom.py +++ b/jax/scipy/stats/geom.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/laplace.py b/jax/scipy/stats/laplace.py index f33742ed7ce1..e32c578e836b 100644 --- a/jax/scipy/stats/laplace.py +++ b/jax/scipy/stats/laplace.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/logistic.py b/jax/scipy/stats/logistic.py index 52c5cefe001f..79eee77cae08 100644 --- a/jax/scipy/stats/logistic.py +++ b/jax/scipy/stats/logistic.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/multinomial.py b/jax/scipy/stats/multinomial.py index 2e10229a5892..9f5819cd4f99 100644 --- a/jax/scipy/stats/multinomial.py +++ b/jax/scipy/stats/multinomial.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/multivariate_normal.py b/jax/scipy/stats/multivariate_normal.py index 69127aeb1476..05f3fa3c4955 100644 --- a/jax/scipy/stats/multivariate_normal.py +++ b/jax/scipy/stats/multivariate_normal.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/nbinom.py b/jax/scipy/stats/nbinom.py index 3134e7c65432..83e0ba9fcd4f 100644 --- a/jax/scipy/stats/nbinom.py +++ b/jax/scipy/stats/nbinom.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/norm.py b/jax/scipy/stats/norm.py index ed1014b13d6e..c4f0331be7c4 100644 --- a/jax/scipy/stats/norm.py +++ b/jax/scipy/stats/norm.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/pareto.py b/jax/scipy/stats/pareto.py index 03bbc975d47a..80b7b6916bb3 100644 --- a/jax/scipy/stats/pareto.py +++ b/jax/scipy/stats/pareto.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/poisson.py b/jax/scipy/stats/poisson.py index 6deb1c684b04..2211f4bdd2d3 100644 --- a/jax/scipy/stats/poisson.py +++ b/jax/scipy/stats/poisson.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/t.py b/jax/scipy/stats/t.py index ecf96c828160..10c59562bab4 100644 --- a/jax/scipy/stats/t.py +++ b/jax/scipy/stats/t.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/scipy/stats/uniform.py b/jax/scipy/stats/uniform.py index 171554a25b8b..62381005651f 100644 --- a/jax/scipy/stats/uniform.py +++ b/jax/scipy/stats/uniform.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/stages.py b/jax/stages.py index 45ac2b730007..bcc6f3f2bc46 100644 --- a/jax/stages.py +++ b/jax/stages.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/test_util.py b/jax/test_util.py index f41706d6de4e..3bd5a85715d6 100644 --- a/jax/test_util.py +++ b/jax/test_util.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/tools/BUILD b/jax/tools/BUILD index e5921d293634..79bf32bd83a6 100644 --- a/jax/tools/BUILD +++ b/jax/tools/BUILD @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/tools/__init__.py b/jax/tools/__init__.py index 086a24e64eb4..f2b3b5a03aef 100644 --- a/jax/tools/__init__.py +++ b/jax/tools/__init__.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/tools/build_defs.bzl b/jax/tools/build_defs.bzl index 3168c763c7e4..7bc54f600bcb 100644 --- a/jax/tools/build_defs.bzl +++ b/jax/tools/build_defs.bzl @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/tools/colab_tpu.py b/jax/tools/colab_tpu.py index c16db7021610..3924a59801d4 100644 --- a/jax/tools/colab_tpu.py +++ b/jax/tools/colab_tpu.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/tools/jax_to_ir.py b/jax/tools/jax_to_ir.py index 753d1f1c460b..752317b0c24f 100644 --- a/jax/tools/jax_to_ir.py +++ b/jax/tools/jax_to_ir.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/tree_util.py b/jax/tree_util.py index 19437de2d813..2cd3b91d2907 100644 --- a/jax/tree_util.py +++ b/jax/tree_util.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/util.py b/jax/util.py index 3cc345017c1b..53e949232520 100644 --- a/jax/util.py +++ b/jax/util.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jax/version.py b/jax/version.py index 4769e008e9ca..a6a1c276c0b9 100644 --- a/jax/version.py +++ b/jax/version.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jaxlib/BUILD b/jaxlib/BUILD index e8235fa2315e..5a6a579e51cc 100644 --- a/jaxlib/BUILD +++ b/jaxlib/BUILD @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jaxlib/cpu_feature_guard.c b/jaxlib/cpu_feature_guard.c index b1be8e8ccf2a..b7fe688eaf52 100644 --- a/jaxlib/cpu_feature_guard.c +++ b/jaxlib/cpu_feature_guard.c @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cpu_kernels.cc b/jaxlib/cpu_kernels.cc index 56f474168a5a..7e8f0fc56ffb 100644 --- a/jaxlib/cpu_kernels.cc +++ b/jaxlib/cpu_kernels.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/BUILD b/jaxlib/cuda/BUILD index 1808918aae8d..37a3f13f6258 100644 --- a/jaxlib/cuda/BUILD +++ b/jaxlib/cuda/BUILD @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cublas.cc b/jaxlib/cuda/cublas.cc index c65e82caa61f..d3539bcd1977 100644 --- a/jaxlib/cuda/cublas.cc +++ b/jaxlib/cuda/cublas.cc @@ -1,4 +1,4 @@ -/* Copyright 2019 Google LLC +/* Copyright 2019 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cublas_kernels.cc b/jaxlib/cuda/cublas_kernels.cc index 6e841fb2fda8..2669ae06c577 100644 --- a/jaxlib/cuda/cublas_kernels.cc +++ b/jaxlib/cuda/cublas_kernels.cc @@ -1,4 +1,4 @@ -/* Copyright 2019 Google LLC +/* Copyright 2019 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cublas_kernels.h b/jaxlib/cuda/cublas_kernels.h index b7dd3886cedc..0fcc8c1eb32f 100644 --- a/jaxlib/cuda/cublas_kernels.h +++ b/jaxlib/cuda/cublas_kernels.h @@ -1,4 +1,4 @@ -/* Copyright 2019 Google LLC +/* Copyright 2019 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cuda_gpu_kernel_helpers.cc b/jaxlib/cuda/cuda_gpu_kernel_helpers.cc index 89b6f1489f47..5887b6034968 100644 --- a/jaxlib/cuda/cuda_gpu_kernel_helpers.cc +++ b/jaxlib/cuda/cuda_gpu_kernel_helpers.cc @@ -1,4 +1,4 @@ -/* Copyright 2019 Google LLC +/* Copyright 2019 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cuda_gpu_kernel_helpers.h b/jaxlib/cuda/cuda_gpu_kernel_helpers.h index 1a0098ff98dc..927d1ea7fd8b 100644 --- a/jaxlib/cuda/cuda_gpu_kernel_helpers.h +++ b/jaxlib/cuda/cuda_gpu_kernel_helpers.h @@ -1,4 +1,4 @@ -/* Copyright 2019 Google LLC +/* Copyright 2019 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cuda_gpu_kernels.cc b/jaxlib/cuda/cuda_gpu_kernels.cc index c323787f02f1..37aa49f46f80 100644 --- a/jaxlib/cuda/cuda_gpu_kernels.cc +++ b/jaxlib/cuda/cuda_gpu_kernels.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cuda_kernels.cc b/jaxlib/cuda/cuda_kernels.cc index f65d8d997187..935fda53c26c 100644 --- a/jaxlib/cuda/cuda_kernels.cc +++ b/jaxlib/cuda/cuda_kernels.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cuda_linalg.cc b/jaxlib/cuda/cuda_linalg.cc index 09afb6b60d12..b798c5bbedc6 100644 --- a/jaxlib/cuda/cuda_linalg.cc +++ b/jaxlib/cuda/cuda_linalg.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cuda_lu_pivot_kernels.cc b/jaxlib/cuda/cuda_lu_pivot_kernels.cc index b13ba14af862..06c0dfd3dad8 100644 --- a/jaxlib/cuda/cuda_lu_pivot_kernels.cc +++ b/jaxlib/cuda/cuda_lu_pivot_kernels.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cuda_lu_pivot_kernels.cu.cc b/jaxlib/cuda/cuda_lu_pivot_kernels.cu.cc index 336d3873eb29..a26fb55f37ff 100644 --- a/jaxlib/cuda/cuda_lu_pivot_kernels.cu.cc +++ b/jaxlib/cuda/cuda_lu_pivot_kernels.cu.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cuda_lu_pivot_kernels.h b/jaxlib/cuda/cuda_lu_pivot_kernels.h index 993c1aa7ec42..673672dcbda3 100644 --- a/jaxlib/cuda/cuda_lu_pivot_kernels.h +++ b/jaxlib/cuda/cuda_lu_pivot_kernels.h @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cuda_prng.cc b/jaxlib/cuda/cuda_prng.cc index c54f4a829982..fd9ca1246e48 100644 --- a/jaxlib/cuda/cuda_prng.cc +++ b/jaxlib/cuda/cuda_prng.cc @@ -1,4 +1,4 @@ -/* Copyright 2019 Google LLC +/* Copyright 2019 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cuda_prng_kernels.cc b/jaxlib/cuda/cuda_prng_kernels.cc index ead6be35aef4..a004b2a20399 100644 --- a/jaxlib/cuda/cuda_prng_kernels.cc +++ b/jaxlib/cuda/cuda_prng_kernels.cc @@ -1,4 +1,4 @@ -/* Copyright 2019 Google LLC +/* Copyright 2019 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cuda_prng_kernels.cu.cc b/jaxlib/cuda/cuda_prng_kernels.cu.cc index ff957414eb7f..6eb87fe2fc78 100644 --- a/jaxlib/cuda/cuda_prng_kernels.cu.cc +++ b/jaxlib/cuda/cuda_prng_kernels.cu.cc @@ -1,4 +1,4 @@ -/* Copyright 2019 Google LLC +/* Copyright 2019 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cuda_prng_kernels.h b/jaxlib/cuda/cuda_prng_kernels.h index 2c1ffbe6a196..035f1f05f27c 100644 --- a/jaxlib/cuda/cuda_prng_kernels.h +++ b/jaxlib/cuda/cuda_prng_kernels.h @@ -1,4 +1,4 @@ -/* Copyright 2019 Google LLC +/* Copyright 2019 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cusolver.cc b/jaxlib/cuda/cusolver.cc index 3da84f9847d8..7bb2a8ed5f1a 100644 --- a/jaxlib/cuda/cusolver.cc +++ b/jaxlib/cuda/cusolver.cc @@ -1,4 +1,4 @@ -/* Copyright 2019 Google LLC +/* Copyright 2019 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cusolver_kernels.cc b/jaxlib/cuda/cusolver_kernels.cc index 7ff986617acb..7979eaceb1a4 100644 --- a/jaxlib/cuda/cusolver_kernels.cc +++ b/jaxlib/cuda/cusolver_kernels.cc @@ -1,4 +1,4 @@ -/* Copyright 2019 Google LLC +/* Copyright 2019 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cusolver_kernels.h b/jaxlib/cuda/cusolver_kernels.h index a886093980d2..3bc0a097e204 100644 --- a/jaxlib/cuda/cusolver_kernels.h +++ b/jaxlib/cuda/cusolver_kernels.h @@ -1,4 +1,4 @@ -/* Copyright 2019 Google LLC +/* Copyright 2019 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cusparse.cc b/jaxlib/cuda/cusparse.cc index 664013b90059..0deb0b8b06b0 100644 --- a/jaxlib/cuda/cusparse.cc +++ b/jaxlib/cuda/cusparse.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cusparse_kernels.cc b/jaxlib/cuda/cusparse_kernels.cc index 0ef226bacda2..364c999479fc 100644 --- a/jaxlib/cuda/cusparse_kernels.cc +++ b/jaxlib/cuda/cusparse_kernels.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/cuda/cusparse_kernels.h b/jaxlib/cuda/cusparse_kernels.h index 8129de8235ce..de644b525121 100644 --- a/jaxlib/cuda/cusparse_kernels.h +++ b/jaxlib/cuda/cusparse_kernels.h @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/ducc_fft.cc b/jaxlib/ducc_fft.cc index fef8d129aed6..b5211667d776 100644 --- a/jaxlib/ducc_fft.cc +++ b/jaxlib/ducc_fft.cc @@ -1,4 +1,4 @@ -/* Copyright 2020 Google LLC +/* Copyright 2020 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/ducc_fft.fbs b/jaxlib/ducc_fft.fbs index 2951d395483b..dcc74c360520 100644 --- a/jaxlib/ducc_fft.fbs +++ b/jaxlib/ducc_fft.fbs @@ -1,4 +1,4 @@ -/* Copyright 2020 Google LLC +/* Copyright 2020 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/ducc_fft.py b/jaxlib/ducc_fft.py index c9f76ab1d062..f805234fee5f 100644 --- a/jaxlib/ducc_fft.py +++ b/jaxlib/ducc_fft.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jaxlib/ducc_fft_kernels.cc b/jaxlib/ducc_fft_kernels.cc index 45012da52ce2..6bed7ffbfebf 100644 --- a/jaxlib/ducc_fft_kernels.cc +++ b/jaxlib/ducc_fft_kernels.cc @@ -1,4 +1,4 @@ -/* Copyright 2020 Google LLC +/* Copyright 2020 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/ducc_fft_kernels.h b/jaxlib/ducc_fft_kernels.h index 1760921b00fe..2491e7e7a886 100644 --- a/jaxlib/ducc_fft_kernels.h +++ b/jaxlib/ducc_fft_kernels.h @@ -1,4 +1,4 @@ -/* Copyright 2020 Google LLC +/* Copyright 2020 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/gpu_linalg.py b/jaxlib/gpu_linalg.py index 771b4f4a762a..a9c432da7cb6 100644 --- a/jaxlib/gpu_linalg.py +++ b/jaxlib/gpu_linalg.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jaxlib/gpu_prng.py b/jaxlib/gpu_prng.py index 2248263cfeee..d7f5965deb32 100644 --- a/jaxlib/gpu_prng.py +++ b/jaxlib/gpu_prng.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jaxlib/gpu_solver.py b/jaxlib/gpu_solver.py index a248ce2af94f..a09fee1862d8 100644 --- a/jaxlib/gpu_solver.py +++ b/jaxlib/gpu_solver.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jaxlib/gpu_sparse.py b/jaxlib/gpu_sparse.py index 7ba5f9479af1..1299953a5872 100644 --- a/jaxlib/gpu_sparse.py +++ b/jaxlib/gpu_sparse.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jaxlib/handle_pool.h b/jaxlib/handle_pool.h index ce1cfb40ea37..a80b9fe42197 100644 --- a/jaxlib/handle_pool.h +++ b/jaxlib/handle_pool.h @@ -1,4 +1,4 @@ -/* Copyright 2019 Google LLC +/* Copyright 2019 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/init.py b/jaxlib/init.py index 95c2c2d4ab56..e1404565419b 100644 --- a/jaxlib/init.py +++ b/jaxlib/init.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jaxlib/jax.bzl b/jaxlib/jax.bzl index 53350e46548f..c94acc8b8e8b 100644 --- a/jaxlib/jax.bzl +++ b/jaxlib/jax.bzl @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jaxlib/kernel_helpers.h b/jaxlib/kernel_helpers.h index 192483765201..33eaf8a1d00b 100644 --- a/jaxlib/kernel_helpers.h +++ b/jaxlib/kernel_helpers.h @@ -1,4 +1,4 @@ -/* Copyright 2019 Google LLC +/* Copyright 2019 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/kernel_pybind11_helpers.h b/jaxlib/kernel_pybind11_helpers.h index db15d7a94bc8..0230aeced95c 100644 --- a/jaxlib/kernel_pybind11_helpers.h +++ b/jaxlib/kernel_pybind11_helpers.h @@ -1,4 +1,4 @@ -/* Copyright 2019 Google LLC +/* Copyright 2019 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/lapack.cc b/jaxlib/lapack.cc index 54b564ddeec9..3824c949f013 100644 --- a/jaxlib/lapack.cc +++ b/jaxlib/lapack.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/lapack.py b/jaxlib/lapack.py index b2abb352fd11..dc94a039ae37 100644 --- a/jaxlib/lapack.py +++ b/jaxlib/lapack.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jaxlib/lapack_kernels.cc b/jaxlib/lapack_kernels.cc index 4ec4c33acebd..68453b045c47 100644 --- a/jaxlib/lapack_kernels.cc +++ b/jaxlib/lapack_kernels.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/lapack_kernels.h b/jaxlib/lapack_kernels.h index 03cf73cc8283..17ac6e160afa 100644 --- a/jaxlib/lapack_kernels.h +++ b/jaxlib/lapack_kernels.h @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/lapack_kernels_using_lapack.cc b/jaxlib/lapack_kernels_using_lapack.cc index 4e77539a8b0e..908e8cdfeb63 100644 --- a/jaxlib/lapack_kernels_using_lapack.cc +++ b/jaxlib/lapack_kernels_using_lapack.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/mhlo_helpers.py b/jaxlib/mhlo_helpers.py index f10b70bea3df..aa6d62dceb7b 100644 --- a/jaxlib/mhlo_helpers.py +++ b/jaxlib/mhlo_helpers.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jaxlib/mlir/BUILD.bazel b/jaxlib/mlir/BUILD.bazel index bb502ab7bec5..4219f5806825 100644 --- a/jaxlib/mlir/BUILD.bazel +++ b/jaxlib/mlir/BUILD.bazel @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jaxlib/mlir/_mlir_libs/BUILD.bazel b/jaxlib/mlir/_mlir_libs/BUILD.bazel index 9417b8ee786d..d213f485ac77 100644 --- a/jaxlib/mlir/_mlir_libs/BUILD.bazel +++ b/jaxlib/mlir/_mlir_libs/BUILD.bazel @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/BUILD.bazel b/jaxlib/rocm/BUILD.bazel index dc92c2e10452..e4344cfa6707 100644 --- a/jaxlib/rocm/BUILD.bazel +++ b/jaxlib/rocm/BUILD.bazel @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hip_gpu_kernel_helpers.cc b/jaxlib/rocm/hip_gpu_kernel_helpers.cc index de5ff6aeb9cc..722b9b584da4 100644 --- a/jaxlib/rocm/hip_gpu_kernel_helpers.cc +++ b/jaxlib/rocm/hip_gpu_kernel_helpers.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hip_gpu_kernel_helpers.h b/jaxlib/rocm/hip_gpu_kernel_helpers.h index 727d4a5ac2bd..42bd4b11b649 100644 --- a/jaxlib/rocm/hip_gpu_kernel_helpers.h +++ b/jaxlib/rocm/hip_gpu_kernel_helpers.h @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hip_linalg.cc b/jaxlib/rocm/hip_linalg.cc index f6116c5850dc..b891f256b8f5 100644 --- a/jaxlib/rocm/hip_linalg.cc +++ b/jaxlib/rocm/hip_linalg.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hip_lu_pivot_kernels.cc b/jaxlib/rocm/hip_lu_pivot_kernels.cc index a078dfff3d75..00f642c14065 100644 --- a/jaxlib/rocm/hip_lu_pivot_kernels.cc +++ b/jaxlib/rocm/hip_lu_pivot_kernels.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hip_lu_pivot_kernels.h b/jaxlib/rocm/hip_lu_pivot_kernels.h index 23317fb02a7a..c75dd7523c2e 100644 --- a/jaxlib/rocm/hip_lu_pivot_kernels.h +++ b/jaxlib/rocm/hip_lu_pivot_kernels.h @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hip_lu_pivot_kernels.hip.cc b/jaxlib/rocm/hip_lu_pivot_kernels.hip.cc index 34eabde7fad5..cf171ee15981 100644 --- a/jaxlib/rocm/hip_lu_pivot_kernels.hip.cc +++ b/jaxlib/rocm/hip_lu_pivot_kernels.hip.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hip_prng.cc b/jaxlib/rocm/hip_prng.cc index 667954935bf6..ba7784c52de2 100644 --- a/jaxlib/rocm/hip_prng.cc +++ b/jaxlib/rocm/hip_prng.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hip_prng_kernels.cc b/jaxlib/rocm/hip_prng_kernels.cc index 05992467d745..a347c0276ce8 100644 --- a/jaxlib/rocm/hip_prng_kernels.cc +++ b/jaxlib/rocm/hip_prng_kernels.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hip_prng_kernels.h b/jaxlib/rocm/hip_prng_kernels.h index f4d503d77a8f..4bc17253a46d 100644 --- a/jaxlib/rocm/hip_prng_kernels.h +++ b/jaxlib/rocm/hip_prng_kernels.h @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hip_prng_kernels.hip.cc b/jaxlib/rocm/hip_prng_kernels.hip.cc index af18739b3512..080195e3df78 100644 --- a/jaxlib/rocm/hip_prng_kernels.hip.cc +++ b/jaxlib/rocm/hip_prng_kernels.hip.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hipblas.cc b/jaxlib/rocm/hipblas.cc index edf4a11f9c0d..3bb093cbb912 100644 --- a/jaxlib/rocm/hipblas.cc +++ b/jaxlib/rocm/hipblas.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hipblas_kernels.cc b/jaxlib/rocm/hipblas_kernels.cc index 3811711380df..948e7dad8d51 100644 --- a/jaxlib/rocm/hipblas_kernels.cc +++ b/jaxlib/rocm/hipblas_kernels.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hipblas_kernels.h b/jaxlib/rocm/hipblas_kernels.h index 01f336789e11..1b7f691eddd9 100644 --- a/jaxlib/rocm/hipblas_kernels.h +++ b/jaxlib/rocm/hipblas_kernels.h @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hipsolver.cc b/jaxlib/rocm/hipsolver.cc index 80a588a67867..d6555c3ba081 100644 --- a/jaxlib/rocm/hipsolver.cc +++ b/jaxlib/rocm/hipsolver.cc @@ -1,4 +1,4 @@ -/* Copyright 2019 Google LLC +/* Copyright 2019 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hipsolver_kernels.cc b/jaxlib/rocm/hipsolver_kernels.cc index aa2b7c9f2c17..a33169f2babf 100644 --- a/jaxlib/rocm/hipsolver_kernels.cc +++ b/jaxlib/rocm/hipsolver_kernels.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hipsolver_kernels.h b/jaxlib/rocm/hipsolver_kernels.h index 08bed80712bc..d1377358ff70 100644 --- a/jaxlib/rocm/hipsolver_kernels.h +++ b/jaxlib/rocm/hipsolver_kernels.h @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hipsparse.cc b/jaxlib/rocm/hipsparse.cc index 58b15537c925..57f3cbb07f36 100644 --- a/jaxlib/rocm/hipsparse.cc +++ b/jaxlib/rocm/hipsparse.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hipsparse_kernels.cc b/jaxlib/rocm/hipsparse_kernels.cc index 5fc01f2cf3d3..b9e061c9d866 100644 --- a/jaxlib/rocm/hipsparse_kernels.cc +++ b/jaxlib/rocm/hipsparse_kernels.cc @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/rocm/hipsparse_kernels.h b/jaxlib/rocm/hipsparse_kernels.h index bce846000059..eef866a26471 100644 --- a/jaxlib/rocm/hipsparse_kernels.h +++ b/jaxlib/rocm/hipsparse_kernels.h @@ -1,4 +1,4 @@ -/* Copyright 2021 Google LLC +/* Copyright 2021 The JAX Authors. Licensed under the Apache License, Version 2.0 (the "License"); you may not use this file except in compliance with the License. diff --git a/jaxlib/setup.py b/jaxlib/setup.py index 86ecac0caa5c..242226e2a519 100644 --- a/jaxlib/setup.py +++ b/jaxlib/setup.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/jaxlib/symlink_files.bzl b/jaxlib/symlink_files.bzl index 0c1adce433ce..203b66022926 100644 --- a/jaxlib/symlink_files.bzl +++ b/jaxlib/symlink_files.bzl @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/setup.py b/setup.py index b0874f853c8d..d38319b22470 100644 --- a/setup.py +++ b/setup.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/BUILD b/tests/BUILD index 7456446d60da..145a6cc63cd8 100644 --- a/tests/BUILD +++ b/tests/BUILD @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/ann_test.py b/tests/ann_test.py index 42c7b3961f27..7e7ab1b2e649 100644 --- a/tests/ann_test.py +++ b/tests/ann_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/api_test.py b/tests/api_test.py index 74e3402670d9..cd0e981a2401 100644 --- a/tests/api_test.py +++ b/tests/api_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/api_util_test.py b/tests/api_util_test.py index ece4177764af..6500f6c943d8 100644 --- a/tests/api_util_test.py +++ b/tests/api_util_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/array_interoperability_test.py b/tests/array_interoperability_test.py index 8c714a27aad2..5ef8a8fc6163 100644 --- a/tests/array_interoperability_test.py +++ b/tests/array_interoperability_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/array_test.py b/tests/array_test.py index 86fd17574a29..043181e06474 100644 --- a/tests/array_test.py +++ b/tests/array_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/batching_test.py b/tests/batching_test.py index 50ac8fe68f45..93f19df763f5 100644 --- a/tests/batching_test.py +++ b/tests/batching_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/benchmarks/xla.py b/tests/benchmarks/xla.py index b5c2f9dc1e6d..868889eaf857 100644 --- a/tests/benchmarks/xla.py +++ b/tests/benchmarks/xla.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/callback_test.py b/tests/callback_test.py index 94acf9868d38..3503f5df7dc7 100644 --- a/tests/callback_test.py +++ b/tests/callback_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/checkify_test.py b/tests/checkify_test.py index 68b839ae0280..16d823fdea9c 100644 --- a/tests/checkify_test.py +++ b/tests/checkify_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/clear_backends_test.py b/tests/clear_backends_test.py index 446b790b75ec..cd138e82c294 100644 --- a/tests/clear_backends_test.py +++ b/tests/clear_backends_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/compilation_cache_test.py b/tests/compilation_cache_test.py index 7585b2fd4ab7..bf959dd54335 100644 --- a/tests/compilation_cache_test.py +++ b/tests/compilation_cache_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/core_test.py b/tests/core_test.py index 9f51c78935a3..9cfd2a15000d 100644 --- a/tests/core_test.py +++ b/tests/core_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/custom_linear_solve_test.py b/tests/custom_linear_solve_test.py index 326e0bf49f3a..c871e8c8b83d 100644 --- a/tests/custom_linear_solve_test.py +++ b/tests/custom_linear_solve_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/custom_object_test.py b/tests/custom_object_test.py index 47311b800c9f..7258ac9231e9 100644 --- a/tests/custom_object_test.py +++ b/tests/custom_object_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/custom_root_test.py b/tests/custom_root_test.py index a32e17597b44..4a1b99211739 100644 --- a/tests/custom_root_test.py +++ b/tests/custom_root_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/debug_nans_test.py b/tests/debug_nans_test.py index 70d04c4e52a1..22f02a6bc704 100644 --- a/tests/debug_nans_test.py +++ b/tests/debug_nans_test.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/debugger_test.py b/tests/debugger_test.py index 3ed91f820928..2e83c254d455 100644 --- a/tests/debugger_test.py +++ b/tests/debugger_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/debugging_primitives_test.py b/tests/debugging_primitives_test.py index 0c73cfc0251b..bf8e29bd1fc5 100644 --- a/tests/debugging_primitives_test.py +++ b/tests/debugging_primitives_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/dtypes_test.py b/tests/dtypes_test.py index 14cf48b37797..716823519b6f 100644 --- a/tests/dtypes_test.py +++ b/tests/dtypes_test.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/dynamic_api_test.py b/tests/dynamic_api_test.py index 81abc7eb1a47..fffc86790b4b 100644 --- a/tests/dynamic_api_test.py +++ b/tests/dynamic_api_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/errors_test.py b/tests/errors_test.py index 56a041a7839b..3424981e4db9 100644 --- a/tests/errors_test.py +++ b/tests/errors_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/fft_test.py b/tests/fft_test.py index 190ef70bbc46..54323b82e943 100644 --- a/tests/fft_test.py +++ b/tests/fft_test.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/filecheck/array.filecheck.py b/tests/filecheck/array.filecheck.py index 7544ee3e2775..d22faa243b9b 100644 --- a/tests/filecheck/array.filecheck.py +++ b/tests/filecheck/array.filecheck.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/filecheck/jax_filecheck_helpers.py b/tests/filecheck/jax_filecheck_helpers.py index f39b4e3be106..f5837d980eef 100644 --- a/tests/filecheck/jax_filecheck_helpers.py +++ b/tests/filecheck/jax_filecheck_helpers.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/filecheck/math.filecheck.py b/tests/filecheck/math.filecheck.py index 932c5465c01b..86778448d10a 100644 --- a/tests/filecheck/math.filecheck.py +++ b/tests/filecheck/math.filecheck.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/filecheck/names.filecheck.py b/tests/filecheck/names.filecheck.py index ebfef6b03086..e90711f1456e 100644 --- a/tests/filecheck/names.filecheck.py +++ b/tests/filecheck/names.filecheck.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/filecheck/shapes.filecheck.py b/tests/filecheck/shapes.filecheck.py index 7d2699f4da9e..be610e8e7c40 100644 --- a/tests/filecheck/shapes.filecheck.py +++ b/tests/filecheck/shapes.filecheck.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/filecheck/subcomputations.filecheck.py b/tests/filecheck/subcomputations.filecheck.py index 2a0b4f2c5d65..2ddc8d371cdc 100644 --- a/tests/filecheck/subcomputations.filecheck.py +++ b/tests/filecheck/subcomputations.filecheck.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/for_loop_test.py b/tests/for_loop_test.py index 5922af03121a..a4c8ddfc9a05 100644 --- a/tests/for_loop_test.py +++ b/tests/for_loop_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/generated_fun_test.py b/tests/generated_fun_test.py index 14389b323a02..c61e268ac0ba 100644 --- a/tests/generated_fun_test.py +++ b/tests/generated_fun_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/gfile_cache_test.py b/tests/gfile_cache_test.py index 8f3bf4aacff2..edcbd4a07947 100644 --- a/tests/gfile_cache_test.py +++ b/tests/gfile_cache_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/global_device_array_test.py b/tests/global_device_array_test.py index d862e23cf096..3fec86c1bda9 100644 --- a/tests/global_device_array_test.py +++ b/tests/global_device_array_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/heap_profiler_test.py b/tests/heap_profiler_test.py index 126a19b4155e..b24f384c0972 100644 --- a/tests/heap_profiler_test.py +++ b/tests/heap_profiler_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/host_callback_test.py b/tests/host_callback_test.py index eb9dfeefa9c2..179a9a72f2d5 100644 --- a/tests/host_callback_test.py +++ b/tests/host_callback_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/host_callback_to_tf_test.py b/tests/host_callback_to_tf_test.py index e186975be1a7..783654275ad6 100644 --- a/tests/host_callback_to_tf_test.py +++ b/tests/host_callback_to_tf_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/image_test.py b/tests/image_test.py index e6b92fd07f25..84d3f0b45a2e 100644 --- a/tests/image_test.py +++ b/tests/image_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/infeed_test.py b/tests/infeed_test.py index e31a161f8eba..b95e8bcfb2ca 100644 --- a/tests/infeed_test.py +++ b/tests/infeed_test.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/jax_jit_test.py b/tests/jax_jit_test.py index 2eb769adc6b4..7fa3cc62cdd6 100644 --- a/tests/jax_jit_test.py +++ b/tests/jax_jit_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/jax_to_ir_test.py b/tests/jax_to_ir_test.py index e0cb96cf5533..5dbbb35d1c10 100644 --- a/tests/jax_to_ir_test.py +++ b/tests/jax_to_ir_test.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/jaxpr_effects_test.py b/tests/jaxpr_effects_test.py index 9ab6490765b2..ca173ce0c3c3 100644 --- a/tests/jaxpr_effects_test.py +++ b/tests/jaxpr_effects_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/jaxpr_util_test.py b/tests/jaxpr_util_test.py index 6d5a61b8da3a..f17e439bed87 100644 --- a/tests/jaxpr_util_test.py +++ b/tests/jaxpr_util_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/jet_test.py b/tests/jet_test.py index 71b4ef12a40b..7b277dec4f35 100644 --- a/tests/jet_test.py +++ b/tests/jet_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/lax_autodiff_test.py b/tests/lax_autodiff_test.py index 67f9c8db41fc..92384f00c00c 100644 --- a/tests/lax_autodiff_test.py +++ b/tests/lax_autodiff_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/lax_control_flow_test.py b/tests/lax_control_flow_test.py index 137d13ff6030..848d44d1664b 100644 --- a/tests/lax_control_flow_test.py +++ b/tests/lax_control_flow_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/lax_numpy_einsum_test.py b/tests/lax_numpy_einsum_test.py index 4e81f089e6bc..be4757e7878b 100644 --- a/tests/lax_numpy_einsum_test.py +++ b/tests/lax_numpy_einsum_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/lax_numpy_indexing_test.py b/tests/lax_numpy_indexing_test.py index e5c72e2165eb..4952c7371d3a 100644 --- a/tests/lax_numpy_indexing_test.py +++ b/tests/lax_numpy_indexing_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/lax_numpy_test.py b/tests/lax_numpy_test.py index 7bbb1c898588..2c97e5db766e 100644 --- a/tests/lax_numpy_test.py +++ b/tests/lax_numpy_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/lax_numpy_vectorize_test.py b/tests/lax_numpy_vectorize_test.py index a792801ce8b8..d5434cd85aed 100644 --- a/tests/lax_numpy_vectorize_test.py +++ b/tests/lax_numpy_vectorize_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/lax_scipy_sparse_test.py b/tests/lax_scipy_sparse_test.py index 93d29a0f7100..d2a36c975149 100644 --- a/tests/lax_scipy_sparse_test.py +++ b/tests/lax_scipy_sparse_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/lax_scipy_test.py b/tests/lax_scipy_test.py index 401f2a164a8a..4454a697af6d 100644 --- a/tests/lax_scipy_test.py +++ b/tests/lax_scipy_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/lax_test.py b/tests/lax_test.py index 4782551dbd91..36b01d91f9b9 100644 --- a/tests/lax_test.py +++ b/tests/lax_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/lax_vmap_test.py b/tests/lax_vmap_test.py index 13cbff2defd9..e27cd3205720 100644 --- a/tests/lax_vmap_test.py +++ b/tests/lax_vmap_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/linalg_test.py b/tests/linalg_test.py index 751c4736f7b0..c08327ee3745 100644 --- a/tests/linalg_test.py +++ b/tests/linalg_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/lobpcg_test.py b/tests/lobpcg_test.py index eb91fb391815..d694df85e749 100644 --- a/tests/lobpcg_test.py +++ b/tests/lobpcg_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/metadata_test.py b/tests/metadata_test.py index 2c032c444161..f55bf82dee41 100644 --- a/tests/metadata_test.py +++ b/tests/metadata_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/multi_device_test.py b/tests/multi_device_test.py index c6e5c9f5310b..8dbd654659df 100644 --- a/tests/multi_device_test.py +++ b/tests/multi_device_test.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/multibackend_test.py b/tests/multibackend_test.py index 54409a25740e..e374e61f76f0 100644 --- a/tests/multibackend_test.py +++ b/tests/multibackend_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/multiprocess_gpu_test.py b/tests/multiprocess_gpu_test.py index 66d79a524d2e..45a38ec80294 100644 --- a/tests/multiprocess_gpu_test.py +++ b/tests/multiprocess_gpu_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/name_stack_test.py b/tests/name_stack_test.py index 68c6280dbdb3..912e334acac3 100644 --- a/tests/name_stack_test.py +++ b/tests/name_stack_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/nn_test.py b/tests/nn_test.py index 6fa7f2badfa7..f4e89f345898 100644 --- a/tests/nn_test.py +++ b/tests/nn_test.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/ode_test.py b/tests/ode_test.py index 1cf371aa7ea1..01fc103949f8 100644 --- a/tests/ode_test.py +++ b/tests/ode_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/optimizers_test.py b/tests/optimizers_test.py index 1169fff09f9a..ced5e065396e 100644 --- a/tests/optimizers_test.py +++ b/tests/optimizers_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/pickle_test.py b/tests/pickle_test.py index 1bd2dcc1cd39..5c11e765db53 100644 --- a/tests/pickle_test.py +++ b/tests/pickle_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/pjit_test.py b/tests/pjit_test.py index 66e58ee14325..1c9911d92466 100644 --- a/tests/pjit_test.py +++ b/tests/pjit_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/pmap_test.py b/tests/pmap_test.py index b63d38f70984..274e0177238e 100644 --- a/tests/pmap_test.py +++ b/tests/pmap_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/polynomial_test.py b/tests/polynomial_test.py index bf28ca4ff644..b2bca658766e 100644 --- a/tests/polynomial_test.py +++ b/tests/polynomial_test.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/profiler_test.py b/tests/profiler_test.py index e214282d7580..476e366db80d 100644 --- a/tests/profiler_test.py +++ b/tests/profiler_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/python_callback_test.py b/tests/python_callback_test.py index f85667ae9931..8509c2092d20 100644 --- a/tests/python_callback_test.py +++ b/tests/python_callback_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/qdwh_test.py b/tests/qdwh_test.py index f3b6c74667ed..5a4b7a090fe3 100644 --- a/tests/qdwh_test.py +++ b/tests/qdwh_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/random_test.py b/tests/random_test.py index 2517ca093c1f..c2fbebb372a7 100644 --- a/tests/random_test.py +++ b/tests/random_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/remote_transfer_test.py b/tests/remote_transfer_test.py index 8bb9e6ef1713..5941495ba50f 100644 --- a/tests/remote_transfer_test.py +++ b/tests/remote_transfer_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/scipy_fft_test.py b/tests/scipy_fft_test.py index f91a050b9f07..f5075f360fa5 100644 --- a/tests/scipy_fft_test.py +++ b/tests/scipy_fft_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/scipy_interpolate_test.py b/tests/scipy_interpolate_test.py index 2752ab1da461..78d98b081a99 100644 --- a/tests/scipy_interpolate_test.py +++ b/tests/scipy_interpolate_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/scipy_ndimage_test.py b/tests/scipy_ndimage_test.py index 8d934d14ec92..7e12f84529c2 100644 --- a/tests/scipy_ndimage_test.py +++ b/tests/scipy_ndimage_test.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/scipy_optimize_test.py b/tests/scipy_optimize_test.py index 562168a0e57e..2002397c4566 100644 --- a/tests/scipy_optimize_test.py +++ b/tests/scipy_optimize_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/scipy_signal_test.py b/tests/scipy_signal_test.py index 7eef39a19357..af2d5a10ffef 100644 --- a/tests/scipy_signal_test.py +++ b/tests/scipy_signal_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/scipy_stats_test.py b/tests/scipy_stats_test.py index ef5af34e1ba4..4f22f959576b 100644 --- a/tests/scipy_stats_test.py +++ b/tests/scipy_stats_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/sparse_test.py b/tests/sparse_test.py index 965d0bb3c0b7..7b008e90714a 100644 --- a/tests/sparse_test.py +++ b/tests/sparse_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/sparsify_test.py b/tests/sparsify_test.py index 4005d5fa7b1c..216372daf7c1 100644 --- a/tests/sparsify_test.py +++ b/tests/sparsify_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/stack_test.py b/tests/stack_test.py index 85c28bcc919e..142573cada33 100644 --- a/tests/stack_test.py +++ b/tests/stack_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/state_test.py b/tests/state_test.py index db2952932218..f63e5dbc293c 100644 --- a/tests/state_test.py +++ b/tests/state_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/stax_test.py b/tests/stax_test.py index d80916b504f4..e927a869985a 100644 --- a/tests/stax_test.py +++ b/tests/stax_test.py @@ -1,4 +1,4 @@ -# Copyright 2018 Google LLC +# Copyright 2018 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/svd_test.py b/tests/svd_test.py index 12ee19907bb2..cee1357df966 100644 --- a/tests/svd_test.py +++ b/tests/svd_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/transfer_guard_test.py b/tests/transfer_guard_test.py index 2745ed88e700..bb15aabc7ad1 100644 --- a/tests/transfer_guard_test.py +++ b/tests/transfer_guard_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/tree_util_test.py b/tests/tree_util_test.py index 4a8e255c5211..f31a2127cb12 100644 --- a/tests/tree_util_test.py +++ b/tests/tree_util_test.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/typing_test.py b/tests/typing_test.py index 0711b629a34f..5d2dbd121dc6 100644 --- a/tests/typing_test.py +++ b/tests/typing_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/util_test.py b/tests/util_test.py index e9e568a23d70..19c7de534c59 100644 --- a/tests/util_test.py +++ b/tests/util_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/version_test.py b/tests/version_test.py index 7bfa8cdc654e..f05ca284aee6 100644 --- a/tests/version_test.py +++ b/tests/version_test.py @@ -1,4 +1,4 @@ -# Copyright 2022 Google LLC +# Copyright 2022 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/x64_context_test.py b/tests/x64_context_test.py index ac16929fd388..0fb44680e88d 100644 --- a/tests/x64_context_test.py +++ b/tests/x64_context_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/xla_bridge_test.py b/tests/xla_bridge_test.py index 3556da4fb6ec..012c287a1186 100644 --- a/tests/xla_bridge_test.py +++ b/tests/xla_bridge_test.py @@ -1,4 +1,4 @@ -# Copyright 2019 Google LLC +# Copyright 2019 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/xla_interpreter_test.py b/tests/xla_interpreter_test.py index 648bcd8a2015..5dcabc14e595 100644 --- a/tests/xla_interpreter_test.py +++ b/tests/xla_interpreter_test.py @@ -1,4 +1,4 @@ -# Copyright 2021 Google LLC +# Copyright 2021 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/tests/xmap_test.py b/tests/xmap_test.py index 65a2bd2361c9..735b2560eb5b 100644 --- a/tests/xmap_test.py +++ b/tests/xmap_test.py @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License. diff --git a/third_party/ducc/workspace.bzl b/third_party/ducc/workspace.bzl index 6cb6910f1730..afbdc1f9d967 100644 --- a/third_party/ducc/workspace.bzl +++ b/third_party/ducc/workspace.bzl @@ -1,4 +1,4 @@ -# Copyright 2020 Google LLC +# Copyright 2020 The JAX Authors. # # Licensed under the Apache License, Version 2.0 (the "License"); # you may not use this file except in compliance with the License.