diff --git a/.github/workflows/cat_slurm_logs.py b/.github/workflows/cat_slurm_logs.py
index 50e911284..0479a4723 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 7cf745452..d2290a608 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 030e9d747..93dd1bf0b 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 000000000..7281acf41
--- /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 32e4fe059..60ababad1 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 0195468d1..f066c6fc2 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 726e00d97..ffe7f160e 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 a0f52d8c5..5b400a75e 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 8e8118a4f..889f5a730 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 c37d57153..860580532 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 75df869f3..f10efab99 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 ef4f7896f..eb7a3225f 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 49c530ac5..559368c73 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 16002a484..1c081e004 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 49ee3e88b..caad6f1a5 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 4db584302..ebb65a317 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 c33e7f88b..dbc1926fb 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 7a26ce598..a4b284eeb 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 8fe47592e..47d02f6f0 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 7fa84f82d..c8d50c641 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 1f27cd6d1..78e0be7f2 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 @@ sys.path.insert(0, os.path.abspath('..'))
 # -- 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 6824a2048..40f8a175f 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 74d15fc78..0241b731b 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 00e129969..421fc8e71 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 1492956e5..b29a7754b 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 e02f12215..4e479eec3 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 b0c7da3d7..0a0a82604 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 8c5cfe14b..35ee94a58 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 3a482132d..e6b1980c0 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 6e5cf2539..c008d49ef 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 b3407b359..ce829c5fb 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 479194563..e14b57812 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 bc7fa9b8b..7c5c12ff4 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 731dee31e..cbc2ffdfe 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 8a17adce4..77be39d83 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 59521f87b..472ebe82c 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 3d9ca4752..a0fe8b996 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 cc88086e7..911126799 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 f05858ef4..141be978f 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 196ab1c03..0a9494e3d 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 bb3e08526..369831470 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 ad5b3bf1d..70df4fb3d 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 d7ac714fd..92316b42e 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 30257881a..60afaf5a8 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 577bdaea4..15d7049f1 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 54cf7246b..616503f07 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 e9f9eb24b..86488e11e 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 363641946..3b07f7216 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 62127fa49..fe4f33aef 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 ced20d119..2ab088dc7 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 5a48a14c1..679bc8c04 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 db82c4888..908f688d3 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 37144ade9..64e55084c 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 5fe40937a..76b421f37 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 aa639a00e..2cc933f00 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 59d84c236..36b05fd8b 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 5a3bff057..3ff980dde 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 e2e6b0474..5e367d3fc 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 9c53ecfec..498a975de 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 f200f0abd..12588d756 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 ec823d271..42a3d73ef 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 5a8f0e109..bfa29ed64 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 56728095f..9f12c19bb 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 c88ecf463..39df27280 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 58982416c..3214be054 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 4bcbebd66..273460e32 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 d4c96dd37..24fa573f9 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 3f45995a4..d604ecf9d 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 894b6247d..8d009c472 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 5cfc6a3d0..2c1c07d3d 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 b8359ebd6..78a17938f 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 f85367db6..bc974966f 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 30257881a..60afaf5a8 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 4cf4f7d8e..5a2946b9e 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 d1316e795..1af3f0c20 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 b6bf557f2..a06eea694 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 74223f090..f1d010362 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 fc2efd747..4cf320d93 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 05b6d705b..6af6c666a 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 d466984ff..ed82f9807 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 bd35ce210..22418eaaf 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 61e45605d..a5a8080c1 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 3cfa8d3a4..a4e02322b 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 dc03a4602..57e534aa1 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 d1c7fe9b4..76e9d5549 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 5a28569b5..609e072e1 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 1a2ddab06..335416486 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 2555ad209..c7d398f11 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 d95cfadb0..f3eab7efc 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 be87c1e5a..47babc4db 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 11533c3a5..9750a7c94 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 35165d85e..4ed84a684 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 509ffeff9..ab7fb735b 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 0cd9f04a5..ae78bc2df 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 5408008fe..b081eecf6 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 ad8e6d64b..0e662538f 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 30b347181..27488194d 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 ea68b3aba..06dbc964d 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 7d1babd1e..e3f662738 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 81d892a24..aa9348d3c 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 38e648296..297f1e9d4 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 30257881a..60afaf5a8 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 cd7e3ba7a..e88a1e230 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 f46703edb..c3330ac41 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 b0c7da3d7..0a0a82604 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 f36e9de15..e2a31f168 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 379224d88..ee0a1d0e1 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 d3fb050ea..c35e762b7 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 97827576e..de063d35b 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 caef0ced3..a4c6926ae 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 d1aca3c20..ed483a889 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 48e55a076..00834edfd 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 b92866808..9d774d196 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 67d5290cb..ae68e3613 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 3f4c970a0..0d1d3ac1c 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 765b9d109..88486c655 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 30257881a..60afaf5a8 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 83f251fcc..5a2381ddc 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 9a76e30a4..479484821 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 b2d89ef88..136c6b9e6 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 7cf84f7bd..05aaa8dd0 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 2fa0cbd2f..0213917f8 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 fa2829b22..494c3a0a7 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 30257881a..60afaf5a8 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 bb545f476..37ee0c309 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 94de1e667..ec57a10af 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 459e3e6f1..50893aff4 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 bb545f476..37ee0c309 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 44eaf51e8..2801325fc 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 c5f91aae2..9e66c78ea 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 30257881a..60afaf5a8 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 61b8cb51d..ffe30362c 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 197e3844c..224361d6d 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 0760b5f89..a139d0d75 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 09d8a5f37..e0d4f0435 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 ebf58be9a..c276a813f 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 30257881a..60afaf5a8 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 e9fddd056..1162b2f7a 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 6d1576e64..703376bc5 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 30257881a..60afaf5a8 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 877c99781..ede4a6691 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 b62f108a4..4fdcd3c63 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 f2366ed7d..b260fa808 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 85179c858..3c92c6e7f 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 f74ffd43e..c38e92158 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 cbdfece02..269e8f299 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 bf4a7afb4..785224c9b 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 52e96753a..f7b7ae9db 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 d4b55b050..95744fc34 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 b51c381fc..126acfe6f 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 e0bc8d9b7..2276c4624 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 827ffd3f2..1c03f3a18 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 5d81a2f6d..6657d0f80 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 4d468690e..3ad75f962 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 ffcaa00ad..99e709a25 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 8b4da9dc7..48c19344e 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 952adc082..1ead3aaf7 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 6e8604981..c258446cf 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 c509ab521..bdc0e8cd4 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 4056ebe78..b9484acb2 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 47cb1e1d9..df315326b 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 d59df543d..799a1cb70 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 94d9feb1e..a6d6e1981 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 8e48c357e..55098306f 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 98edd0a7d..96725a1e2 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 ea1b1034d..e4dd42d49 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 d11f7117e..e16d11fcd 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 91f89f321..686390886 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 44c6a347a..51c09e12c 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 3d7e259df..6ef91ff86 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 16ec4f7c4..4cb94ff7a 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 900208b71..926a3584b 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 de8ddf7a2..a9072fb61 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 18da9c4bd..b7413954a 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 2fe9b0025..77b3ae36f 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 0a4337cd8..8cc49ac3b 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 452185dea..c61d1a347 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 8486017b6..b8283200c 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 e585607f1..092c39938 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 61cdd71a3..a4850f04c 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 51f55e0ef..a178d7d4b 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 76a75a771..311139da2 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 bf328a8c8..bd428bdbc 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 2d4720c56..284ae6f95 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 50d03abfa..707e966ee 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 cc7589c77..dc321bdb4 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 b8afb8e87..80775a0c1 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 4fbd93bb2..577c9dcb2 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 239b18a14..46eb7e8a6 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 ba711b886..e5bf38c6a 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 a1510dcbb..7d55b9de0 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 b2b2aed16..1f325cda3 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 33da94adc..ccbc9f348 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 9987d0f10..c72e032ec 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 4fbd93bb2..577c9dcb2 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 5359249bf..81e2fd5a6 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 84dda9f3c..cd3653bc9 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 9c3dc4579..03dcabfdb 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 4fbd93bb2..577c9dcb2 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 1797a2d06..be0ae2b88 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 1242521ae..f3bfd2143 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 4a52f2d1e..1993f450a 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 b55e0bfc8..2ba1f4c8f 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 308a66c67..90315f7ca 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 c83eda7de..6a2dd2fd7 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 1d7b4f267..e9ea8af01 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 4fbd93bb2..577c9dcb2 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 aae32716b..52c3203c6 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 540821dc4..7e282f0f2 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 cbc5bbe56..3e7f6ea1f 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 987cc28d3..63eb6faa2 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 a2eeeb7ea..ed7b8eefd 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 e160e1ad2..de7d21652 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 4fbd93bb2..577c9dcb2 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 63c29a96c..8fd0b6fc6 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 a18e474d6..19c564bda 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 25e30eb30..151610728 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 0cc346e7c..5452ea13d 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 3fb3cd368..5adfed582 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 4021a5492..190433000 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 bbd1e0c38..6e5834709 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 30257881a..60afaf5a8 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 e88c4c81e..65771e595 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 6d87a31fa..72ce456d0 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 10dd7346a..61fcedea7 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 cda8184e6..761c27154 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 57533bf29..b2e537bc6 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 c023c8bb2..022f221cd 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 8e9c82aae..dea8f23a2 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 7bca5cc2c..67cad2c62 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 84243e076..b03b66f65 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 91fce99fa..633240c8a 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 282ee4afe..4ccdd3ab9 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 f2ff61a15..5fd9a0d40 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 9111b0fb8..b620bade1 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 65cbebec7..2a43035e4 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 1fa62480a..51f0a816b 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 e21ac392c..7b8fa4e12 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 29c99491e..d6d8663a5 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 cfdd079ae..65e9225f9 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 efe7173cb..9a67ae35b 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 ea70485a5..868cb5b57 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 c2a57999d..20d1e01ef 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 4f55caa4f..ab5039f4f 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 ff74f774e..6e90b00ff 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 246da8ffd..71e7eef8f 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 990d41658..53ad17ae5 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 f3f36f031..886413ddc 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 8b13fc9c1..22b1ae277 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 75665a083..74e791b7e 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 c9e029029..ff7323644 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 34a7a6a36..dc253974f 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 88c2d96e5..dafb4b3e0 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 4bfaafc8d..1887c4cc0 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 1171f2a4b..a650cd4fb 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 10499d0df..6adaee8f9 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 b312a413c..49551fadc 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 321ee746e..82fb0f89a 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 eaeee8f3e..b5f5cba33 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 26f0c68e6..1d536cca4 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 de08f56c3..da7f6e038 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 d9bf532c3..8b05e82c9 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 43ada03f6..11da9cfda 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 bf8434bbb..5343f3333 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 d011a9667..628eb7b60 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 297f218db..0736659b7 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 6911f2b7d..672de5f1f 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 26d45a07f..4fbd40937 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 93c2b2975..e50e9f151 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 d08e6cd09..f52a15271 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 8502b3c07..1cf335689 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 cf4cd8a8b..8cc65f283 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 6176d860f..405cfe34a 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 bebb5205d..f243a41d4 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 532e1a519..be7591d3f 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 53480a66f..e83300141 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 4bc61fd42..28f4edb94 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 0ea575964..9841d4cb2 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 b1c042091..e7f775e3d 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 d5691d2ac..1d4599a38 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 6fbc55c81..73a106967 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 0ae9cd1a2..cf2b50fce 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 e74c1165c..531a0889e 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 be33b038f..96574086d 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 8fba2c934..f0fe691a3 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 cc181e528..38b8b21d4 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 7cc4d4a5c..1df004ce8 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 d81a77840..246ea9004 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 75ab9215b..1b307dd7f 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 0f504cf64..47f5bc0cf 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 c0c3ced15..1ed987ad9 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 0b836b09c..6966bd919 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 e88567bb6..759c78074 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 17253362a..22cb3d77e 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 1a7b9cee5..dddf6a5b9 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 3d822f08d..ffc4970b8 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 81e162497..b4ca71832 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 4fdec22d1..34cdb7b25 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 42170049f..a14fb441e 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 3c1a0f390..d98236497 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 91694f9df..9c2279ec1 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 f9d749b20..c57bcbb51 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 010835c81..0cbe5d603 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 1c4afecc7..a8024a10e 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 75a1cc257..ab122cad3 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 4429d9bbc..8cdde458a 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 58d26c878..99ca46ab5 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 42153d4b7..86dbd0dc0 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 550551445..19528bb08 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 25962836b..320e4e236 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 2d069773d..c35830a92 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 d69cbebc8..92bb24626 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 f33742ed7..e32c578e8 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 52c5cefe0..79eee77ca 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 2e10229a5..9f5819cd4 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 69127aeb1..05f3fa3c4 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 3134e7c65..83e0ba9fc 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 ed1014b13..c4f0331be 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 03bbc975d..80b7b6916 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 6deb1c684..2211f4bdd 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 ecf96c828..10c59562b 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 171554a25..623810056 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 45ac2b730..bcc6f3f2b 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 f41706d6d..3bd5a8571 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 e5921d293..79bf32bd8 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 086a24e64..f2b3b5a03 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 3168c763c..7bc54f600 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 c16db7021..3924a5980 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 753d1f1c4..752317b0c 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 19437de2d..2cd3b91d2 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 3cc345017..53e949232 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 4769e008e..a6a1c276c 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 e8235fa23..5a6a579e5 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 b1be8e8cc..b7fe688ea 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 56f474168..7e8f0fc56 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 1808918aa..37a3f13f6 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 c65e82caa..d3539bcd1 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 6e841fb2f..2669ae06c 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 b7dd3886c..0fcc8c1eb 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 89b6f1489..5887b6034 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 1a0098ff9..927d1ea7f 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 c323787f0..37aa49f46 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 f65d8d997..935fda53c 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 09afb6b60..b798c5bbe 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 b13ba14af..06c0dfd3d 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 336d3873e..a26fb55f3 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 993c1aa7e..673672dcb 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 c54f4a829..fd9ca1246 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 ead6be35a..a004b2a20 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 ff957414e..6eb87fe2f 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 2c1ffbe6a..035f1f05f 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 3da84f984..7bb2a8ed5 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 7ff986617..7979eaceb 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 a88609398..3bc0a097e 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 664013b90..0deb0b8b0 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 0ef226bac..364c99947 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 8129de823..de644b525 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 fef8d129a..b5211667d 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 2951d3954..dcc74c360 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 c9f76ab1d..f805234fe 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 45012da52..6bed7ffbf 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 1760921b0..2491e7e7a 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 771b4f4a7..a9c432da7 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 2248263cf..d7f5965de 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 a248ce2af..a09fee186 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 7ba5f9479..1299953a5 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 ce1cfb40e..a80b9fe42 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 95c2c2d4a..e14045654 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 53350e465..c94acc8b8 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 192483765..33eaf8a1d 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 db15d7a94..0230aeced 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 54b564dde..3824c949f 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 b2abb352f..dc94a039a 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 4ec4c33ac..68453b045 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 03cf73cc8..17ac6e160 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 4e77539a8..908e8cdfe 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 f10b70bea..aa6d62dce 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 bb502ab7b..4219f5806 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 9417b8ee7..d213f485a 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 dc92c2e10..e4344cfa6 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 de5ff6aeb..722b9b584 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 727d4a5ac..42bd4b11b 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 f6116c585..b891f256b 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 a078dfff3..00f642c14 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 23317fb02..c75dd7523 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 34eabde7f..cf171ee15 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 667954935..ba7784c52 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 05992467d..a347c0276 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 f4d503d77..4bc17253a 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 af18739b3..080195e3d 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 edf4a11f9..3bb093cbb 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 381171138..948e7dad8 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 01f336789..1b7f691ed 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 80a588a67..d6555c3ba 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 aa2b7c9f2..a33169f2b 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 08bed8071..d1377358f 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 58b15537c..57f3cbb07 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 5fc01f2cf..b9e061c9d 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 bce846000..eef866a26 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 86ecac0ca..242226e2a 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 0c1adce43..203b66022 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 b0874f853..d38319b22 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 7456446d6..145a6cc63 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 42c7b3961..7e7ab1b2e 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 74e340267..cd0e981a2 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 ece417776..6500f6c94 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 8c714a27a..5ef8a8fc6 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 86fd17574..043181e06 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 50ac8fe68..93f19df76 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 b5c2f9dc1..868889eaf 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 94acf9868..3503f5df7 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 68b839ae0..16d823fde 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 446b790b7..cd138e82c 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 7585b2fd4..bf959dd54 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 9f51c7893..9cfd2a150 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 326e0bf49..c871e8c8b 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 47311b800..7258ac923 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 a32e17597..4a1b99211 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 70d04c4e5..22f02a6bc 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 3ed91f820..2e83c254d 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 0c73cfc02..bf8e29bd1 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 14cf48b37..716823519 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 81abc7eb1..fffc86790 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 56a041a78..3424981e4 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 190ef70bb..54323b82e 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 7544ee3e2..d22faa243 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 f39b4e3be..f5837d980 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 932c5465c..86778448d 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 ebfef6b03..e90711f14 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 7d2699f4d..be610e8e7 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 2a0b4f2c5..2ddc8d371 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 5922af031..a4c8ddfc9 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 14389b323..c61e268ac 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 8f3bf4aac..edcbd4a07 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 d862e23cf..3fec86c1b 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 126a19b41..b24f384c0 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 eb9dfeefa..179a9a72f 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 e186975be..783654275 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 e6b92fd07..84d3f0b45 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 e31a161f8..b95e8bcfb 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 2eb769adc..7fa3cc62c 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 e0cb96cf5..5dbbb35d1 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 9ab649076..ca173ce0c 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 6d5a61b8d..f17e439be 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 71b4ef12a..7b277dec4 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 67f9c8db4..92384f00c 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 137d13ff6..848d44d16 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 4e81f089e..be4757e78 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 e5c72e216..4952c7371 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 7bbb1c898..2c97e5db7 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 a792801ce..d5434cd85 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 93d29a0f7..d2a36c975 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 401f2a164..4454a697a 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 4782551db..36b01d91f 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 13cbff2de..e27cd3205 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 751c4736f..c08327ee3 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 eb91fb391..d694df85e 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 2c032c444..f55bf82de 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 c6e5c9f53..8dbd65465 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 54409a257..e374e61f7 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 66d79a524..45a38ec80 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 68c6280db..912e334ac 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 6fa7f2bad..f4e89f345 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 1cf371aa7..01fc10394 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 1169fff09..ced5e0653 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 1bd2dcc1c..5c11e765d 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 66e58ee14..1c9911d92 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 b63d38f70..274e01772 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 bf28ca4ff..b2bca6587 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 e214282d7..476e366db 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 f85667ae9..8509c2092 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 f3b6c7466..5a4b7a090 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 2517ca093..c2fbebb37 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 8bb9e6ef1..5941495ba 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 f91a050b9..f5075f360 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 2752ab1da..78d98b081 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 8d934d14e..7e12f8452 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 562168a0e..2002397c4 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 7eef39a19..af2d5a10f 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 ef5af34e1..4f22f9595 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 965d0bb3c..7b008e907 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 4005d5fa7..216372daf 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 85c28bcc9..142573cad 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 db2952932..f63e5dbc2 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 d80916b50..e927a8699 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 12ee19907..cee1357df 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 2745ed88e..bb15aabc7 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 4a8e255c5..f31a2127c 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 0711b629a..5d2dbd121 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 e9e568a23..19c7de534 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 7bfa8cdc6..f05ca284a 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 ac16929fd..0fb44680e 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 3556da4fb..012c287a1 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 648bcd8a2..5dcabc14e 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 65a2bd236..735b2560e 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 6cb6910f1..afbdc1f9d 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.