diff --git a/ACKNOWLEDGMENTS.md b/ACKNOWLEDGMENTS.md
new file mode 100644
index 000000000..93c6eb216
--- /dev/null
+++ b/ACKNOWLEDGMENTS.md
@@ -0,0 +1,30 @@
+MLX leverages several third-party software, listed here together with
+their license copied verbatim.
+
+## PocketFFT
+
+Copyright (C) 2010-2018 Max-Planck-Society
+All rights reserved.
+
+Redistribution and use in source and binary forms, with or without modification,
+are permitted provided that the following conditions are met:
+
+* Redistributions of source code must retain the above copyright notice, this
+  list of conditions and the following disclaimer.
+* Redistributions in binary form must reproduce the above copyright notice, this
+  list of conditions and the following disclaimer in the documentation and/or
+  other materials provided with the distribution.
+* Neither the name of the copyright holder nor the names of its contributors may
+  be used to endorse or promote products derived from this software without
+  specific prior written permission.
+
+THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
+ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
+WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDER OR CONTRIBUTORS BE LIABLE FOR
+ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
+ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
diff --git a/LICENSE b/LICENSE
index bb7f03109..98ff47b9e 100644
--- a/LICENSE
+++ b/LICENSE
@@ -1,6 +1,6 @@
 MIT License
 
-Copyright (c) 2023 ml-explore
+Copyright © 2023 Apple Inc.
 
 Permission is hereby granted, free of charge, to any person obtaining a copy
 of this software and associated documentation files (the "Software"), to deal
diff --git a/benchmarks/cpp/autograd.cpp b/benchmarks/cpp/autograd.cpp
index fee3fc616..3beaa04bc 100644
--- a/benchmarks/cpp/autograd.cpp
+++ b/benchmarks/cpp/autograd.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <iostream>
 
 #include "mlx/mlx.h"
diff --git a/benchmarks/cpp/compare_devices.cpp b/benchmarks/cpp/compare_devices.cpp
index 9e7c696e6..eecbc9c4a 100644
--- a/benchmarks/cpp/compare_devices.cpp
+++ b/benchmarks/cpp/compare_devices.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <iostream>
 #include "mlx/mlx.h"
 #include "time_utils.h"
diff --git a/benchmarks/cpp/irregular_strides.cpp b/benchmarks/cpp/irregular_strides.cpp
index e6a570995..50d4c5b73 100644
--- a/benchmarks/cpp/irregular_strides.cpp
+++ b/benchmarks/cpp/irregular_strides.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <iostream>
 #include <sstream>
 
diff --git a/benchmarks/cpp/single_ops.cpp b/benchmarks/cpp/single_ops.cpp
index aa0fa2b36..e8bc3a83d 100644
--- a/benchmarks/cpp/single_ops.cpp
+++ b/benchmarks/cpp/single_ops.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "mlx/mlx.h"
 #include "time_utils.h"
 
diff --git a/benchmarks/cpp/time_utils.h b/benchmarks/cpp/time_utils.h
index 9a4a0778c..780f1867f 100644
--- a/benchmarks/cpp/time_utils.h
+++ b/benchmarks/cpp/time_utils.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <chrono>
diff --git a/benchmarks/numpy/single_ops.py b/benchmarks/numpy/single_ops.py
index e359d3ec0..2631c50fc 100644
--- a/benchmarks/numpy/single_ops.py
+++ b/benchmarks/numpy/single_ops.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import numpy as np
 
 from time_utils import time_fn
diff --git a/benchmarks/numpy/time_utils.py b/benchmarks/numpy/time_utils.py
index 3741fba1d..163f007dc 100644
--- a/benchmarks/numpy/time_utils.py
+++ b/benchmarks/numpy/time_utils.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import time
 
 
diff --git a/benchmarks/python/batch_matmul_bench.py b/benchmarks/python/batch_matmul_bench.py
index 11f0734eb..84bf95fb0 100644
--- a/benchmarks/python/batch_matmul_bench.py
+++ b/benchmarks/python/batch_matmul_bench.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import argparse
 import mlx.core as mx
 
diff --git a/benchmarks/python/blas/bench_gemm.py b/benchmarks/python/blas/bench_gemm.py
index 27cd0793b..608a7cb95 100644
--- a/benchmarks/python/blas/bench_gemm.py
+++ b/benchmarks/python/blas/bench_gemm.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import numpy as np
 import argparse
 import mlx.core as mx
diff --git a/benchmarks/python/blas/bench_gemv.py b/benchmarks/python/blas/bench_gemv.py
index e95e48fa4..41ac4ec3a 100644
--- a/benchmarks/python/blas/bench_gemv.py
+++ b/benchmarks/python/blas/bench_gemv.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import matplotlib.pyplot as plt
 import numpy as np
 import argparse
diff --git a/benchmarks/python/comparative/bench_mlx.py b/benchmarks/python/comparative/bench_mlx.py
index a89774270..5f2c3619f 100644
--- a/benchmarks/python/comparative/bench_mlx.py
+++ b/benchmarks/python/comparative/bench_mlx.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import argparse
 import math
 import os
diff --git a/benchmarks/python/comparative/bench_torch.py b/benchmarks/python/comparative/bench_torch.py
index 444ae6ce1..f275b9da4 100644
--- a/benchmarks/python/comparative/bench_torch.py
+++ b/benchmarks/python/comparative/bench_torch.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import argparse
 import os
 import time
diff --git a/benchmarks/python/comparative/compare.py b/benchmarks/python/comparative/compare.py
index 00b9e4071..f357165fd 100644
--- a/benchmarks/python/comparative/compare.py
+++ b/benchmarks/python/comparative/compare.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 #!/usr/bin/env python
 
 import argparse
diff --git a/benchmarks/python/llama_jax_bench.py b/benchmarks/python/llama_jax_bench.py
index e7539eca3..9f647426e 100644
--- a/benchmarks/python/llama_jax_bench.py
+++ b/benchmarks/python/llama_jax_bench.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import math
 import time
 
diff --git a/benchmarks/python/llama_mlx_bench.py b/benchmarks/python/llama_mlx_bench.py
index f1cea4735..b26ec24bd 100644
--- a/benchmarks/python/llama_mlx_bench.py
+++ b/benchmarks/python/llama_mlx_bench.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import math
 import time
 
diff --git a/benchmarks/python/llama_torch_bench.py b/benchmarks/python/llama_torch_bench.py
index 25b24484b..72329f373 100644
--- a/benchmarks/python/llama_torch_bench.py
+++ b/benchmarks/python/llama_torch_bench.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import math
 import time
 
diff --git a/benchmarks/python/single_ops.py b/benchmarks/python/single_ops.py
index f2977175b..631eef4e2 100644
--- a/benchmarks/python/single_ops.py
+++ b/benchmarks/python/single_ops.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import argparse
 import mlx.core as mx
 
diff --git a/benchmarks/python/time_utils.py b/benchmarks/python/time_utils.py
index e067c8098..73266cb7a 100644
--- a/benchmarks/python/time_utils.py
+++ b/benchmarks/python/time_utils.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import time
 
 import mlx.core as mx
diff --git a/docs/src/conf.py b/docs/src/conf.py
index 9018c231f..5d3be4e57 100644
--- a/docs/src/conf.py
+++ b/docs/src/conf.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 # -*- coding: utf-8 -*-
 
 import os
diff --git a/examples/cpp/linear_regression.cpp b/examples/cpp/linear_regression.cpp
index e6897b7a7..f921da15b 100644
--- a/examples/cpp/linear_regression.cpp
+++ b/examples/cpp/linear_regression.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <chrono>
 #include <cmath>
 #include <iostream>
diff --git a/examples/cpp/logistic_regression.cpp b/examples/cpp/logistic_regression.cpp
index c34ae7fed..7f13f35bb 100644
--- a/examples/cpp/logistic_regression.cpp
+++ b/examples/cpp/logistic_regression.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <chrono>
 #include <cmath>
 #include <iostream>
diff --git a/examples/cpp/timer.h b/examples/cpp/timer.h
index 5a270f00a..cea85ee82 100644
--- a/examples/cpp/timer.h
+++ b/examples/cpp/timer.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <chrono>
diff --git a/examples/cpp/tutorial.cpp b/examples/cpp/tutorial.cpp
index 8426878e2..5dc0e0472 100644
--- a/examples/cpp/tutorial.cpp
+++ b/examples/cpp/tutorial.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 #include <iostream>
 
diff --git a/examples/extensions/axpby/axpby.cpp b/examples/extensions/axpby/axpby.cpp
index 1e6a995bf..56a09f34e 100644
--- a/examples/extensions/axpby/axpby.cpp
+++ b/examples/extensions/axpby/axpby.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 #include <iostream>
 #include <sstream>
diff --git a/examples/extensions/axpby/axpby.h b/examples/extensions/axpby/axpby.h
index 6abc48a9e..8c25980f4 100644
--- a/examples/extensions/axpby/axpby.h
+++ b/examples/extensions/axpby/axpby.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/ops.h"
diff --git a/examples/extensions/axpby/axpby.metal b/examples/extensions/axpby/axpby.metal
index fa2849579..64980578f 100644
--- a/examples/extensions/axpby/axpby.metal
+++ b/examples/extensions/axpby/axpby.metal
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <metal_stdlib>
 
 #include "mlx/backend/metal/kernels/bf16.h"
diff --git a/examples/extensions/bindings.cpp b/examples/extensions/bindings.cpp
index d2eeca4ac..661ddcbaf 100644
--- a/examples/extensions/bindings.cpp
+++ b/examples/extensions/bindings.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <pybind11/pybind11.h>
 #include <pybind11/stl.h>
 
diff --git a/examples/extensions/mlx_sample_extensions/__init__.py b/examples/extensions/mlx_sample_extensions/__init__.py
index 1b17da245..c194d099c 100644
--- a/examples/extensions/mlx_sample_extensions/__init__.py
+++ b/examples/extensions/mlx_sample_extensions/__init__.py
@@ -1,2 +1,4 @@
+# Copyright © 2023 Apple Inc.
+
 import mlx.core as mx
 from .mlx_sample_extensions import *
diff --git a/examples/extensions/setup.py b/examples/extensions/setup.py
index c3817b049..a9ceb39d4 100644
--- a/examples/extensions/setup.py
+++ b/examples/extensions/setup.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 from mlx import extension
 from setuptools import setup
 
diff --git a/examples/python/linear_regression.py b/examples/python/linear_regression.py
index f1837a648..47b0253d9 100644
--- a/examples/python/linear_regression.py
+++ b/examples/python/linear_regression.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import mlx.core as mx
 import time
 
diff --git a/examples/python/logistic_regression.py b/examples/python/logistic_regression.py
index 928ae34a5..8df1fd994 100644
--- a/examples/python/logistic_regression.py
+++ b/examples/python/logistic_regression.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import mlx.core as mx
 import time
 
diff --git a/mlx/allocator.cpp b/mlx/allocator.cpp
index 1d8cddd22..b591caeeb 100644
--- a/mlx/allocator.cpp
+++ b/mlx/allocator.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cstdlib>
 #include <sstream>
 
diff --git a/mlx/allocator.h b/mlx/allocator.h
index 29421d47c..2c3adadf4 100644
--- a/mlx/allocator.h
+++ b/mlx/allocator.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <cstdlib>
diff --git a/mlx/array.cpp b/mlx/array.cpp
index 20ea2e407..0a7b52a94 100644
--- a/mlx/array.cpp
+++ b/mlx/array.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <functional>
 
 #include "mlx/array.h"
diff --git a/mlx/array.h b/mlx/array.h
index c0f02ee4d..6f26c3314 100644
--- a/mlx/array.h
+++ b/mlx/array.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 #include <algorithm>
 #include <cstdint>
diff --git a/mlx/backend/accelerate/conv.cpp b/mlx/backend/accelerate/conv.cpp
index 432e60c59..22ddd907d 100644
--- a/mlx/backend/accelerate/conv.cpp
+++ b/mlx/backend/accelerate/conv.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 
 #include <simd/vector.h>
diff --git a/mlx/backend/accelerate/matmul.cpp b/mlx/backend/accelerate/matmul.cpp
index 5e8014c69..5a38e3123 100644
--- a/mlx/backend/accelerate/matmul.cpp
+++ b/mlx/backend/accelerate/matmul.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 
 #include <vecLib/BNNS/bnns.h>
diff --git a/mlx/backend/accelerate/primitives.cpp b/mlx/backend/accelerate/primitives.cpp
index 1a2b7b3e0..4cde5bbaf 100644
--- a/mlx/backend/accelerate/primitives.cpp
+++ b/mlx/backend/accelerate/primitives.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 #include <cmath>
 
diff --git a/mlx/backend/accelerate/reduce.cpp b/mlx/backend/accelerate/reduce.cpp
index 19de62f56..db2b8eba2 100644
--- a/mlx/backend/accelerate/reduce.cpp
+++ b/mlx/backend/accelerate/reduce.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 
 #include <simd/vector.h>
diff --git a/mlx/backend/accelerate/softmax.cpp b/mlx/backend/accelerate/softmax.cpp
index 31cdbd2da..fcd8fbe50 100644
--- a/mlx/backend/accelerate/softmax.cpp
+++ b/mlx/backend/accelerate/softmax.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 #include <limits>
 
diff --git a/mlx/backend/accelerate/utils.h b/mlx/backend/accelerate/utils.h
index 7c2f1ae65..1dbabe304 100644
--- a/mlx/backend/accelerate/utils.h
+++ b/mlx/backend/accelerate/utils.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <vecLib/BNNS/bnns.h>
diff --git a/mlx/backend/common/arange.h b/mlx/backend/common/arange.h
index 05d874884..5c155dc09 100644
--- a/mlx/backend/common/arange.h
+++ b/mlx/backend/common/arange.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/allocator.h"
diff --git a/mlx/backend/common/arg_reduce.cpp b/mlx/backend/common/arg_reduce.cpp
index 221df856f..f4a672591 100644
--- a/mlx/backend/common/arg_reduce.cpp
+++ b/mlx/backend/common/arg_reduce.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 
 #include "mlx/primitives.h"
diff --git a/mlx/backend/common/binary.cpp b/mlx/backend/common/binary.cpp
index 3e4166a7a..38f0461cc 100644
--- a/mlx/backend/common/binary.cpp
+++ b/mlx/backend/common/binary.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 #include <cmath>
 #include <sstream>
diff --git a/mlx/backend/common/binary.h b/mlx/backend/common/binary.h
index 744ab5202..4b0631746 100644
--- a/mlx/backend/common/binary.h
+++ b/mlx/backend/common/binary.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/allocator.h"
diff --git a/mlx/backend/common/conv.cpp b/mlx/backend/common/conv.cpp
index 863aa9411..f46a68d6f 100644
--- a/mlx/backend/common/conv.cpp
+++ b/mlx/backend/common/conv.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 
 #ifdef ACCELERATE_NEW_LAPACK
diff --git a/mlx/backend/common/copy.cpp b/mlx/backend/common/copy.cpp
index 36d79ac9d..c8af4b548 100644
--- a/mlx/backend/common/copy.cpp
+++ b/mlx/backend/common/copy.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <numeric>
 
 #include "mlx/allocator.h"
diff --git a/mlx/backend/common/copy.h b/mlx/backend/common/copy.h
index dc46e91b4..0affddec3 100644
--- a/mlx/backend/common/copy.h
+++ b/mlx/backend/common/copy.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/array.h"
diff --git a/mlx/backend/common/default_primitives.cpp b/mlx/backend/common/default_primitives.cpp
index 3ed9f2750..a73853ed9 100644
--- a/mlx/backend/common/default_primitives.cpp
+++ b/mlx/backend/common/default_primitives.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cblas.h>
 
 #include "mlx/array.h"
diff --git a/mlx/backend/common/erf.cpp b/mlx/backend/common/erf.cpp
index 0b876a1b0..83769d078 100644
--- a/mlx/backend/common/erf.cpp
+++ b/mlx/backend/common/erf.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cmath>
 
 namespace mlx::core {
diff --git a/mlx/backend/common/erf.h b/mlx/backend/common/erf.h
index d7698754d..25a53d375 100644
--- a/mlx/backend/common/erf.h
+++ b/mlx/backend/common/erf.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 
 namespace mlx::core {
 
diff --git a/mlx/backend/common/fft.cpp b/mlx/backend/common/fft.cpp
index 385d9cb71..e46fd7f92 100644
--- a/mlx/backend/common/fft.cpp
+++ b/mlx/backend/common/fft.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <numeric>
 
 #include "mlx/3rdparty/pocketfft.h"
diff --git a/mlx/backend/common/indexing.cpp b/mlx/backend/common/indexing.cpp
index 9572c1caf..64fc21ec8 100644
--- a/mlx/backend/common/indexing.cpp
+++ b/mlx/backend/common/indexing.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cassert>
 #include <cmath>
diff --git a/mlx/backend/common/load.cpp b/mlx/backend/common/load.cpp
index 5f5f17329..e68ce7f6f 100644
--- a/mlx/backend/common/load.cpp
+++ b/mlx/backend/common/load.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cassert>
 #include <utility>
diff --git a/mlx/backend/common/primitives.cpp b/mlx/backend/common/primitives.cpp
index 86bc746c6..4049af2eb 100644
--- a/mlx/backend/common/primitives.cpp
+++ b/mlx/backend/common/primitives.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cassert>
 #include <cmath>
diff --git a/mlx/backend/common/reduce.cpp b/mlx/backend/common/reduce.cpp
index 90a466f7d..e9d091e49 100644
--- a/mlx/backend/common/reduce.cpp
+++ b/mlx/backend/common/reduce.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 #include <functional>
 #include <limits>
diff --git a/mlx/backend/common/reduce.h b/mlx/backend/common/reduce.h
index 31b1716c3..740f54a48 100644
--- a/mlx/backend/common/reduce.h
+++ b/mlx/backend/common/reduce.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/backend/common/utils.h"
diff --git a/mlx/backend/common/scan.cpp b/mlx/backend/common/scan.cpp
index 89e09e373..9f994b902 100644
--- a/mlx/backend/common/scan.cpp
+++ b/mlx/backend/common/scan.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 
 #include "mlx/backend/common/copy.h"
diff --git a/mlx/backend/common/softmax.cpp b/mlx/backend/common/softmax.cpp
index dabf81143..90874c72d 100644
--- a/mlx/backend/common/softmax.cpp
+++ b/mlx/backend/common/softmax.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 #include <cmath>
 
diff --git a/mlx/backend/common/sort.cpp b/mlx/backend/common/sort.cpp
index 8f0295591..edeee065b 100644
--- a/mlx/backend/common/sort.cpp
+++ b/mlx/backend/common/sort.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cassert>
 #include <cmath>
diff --git a/mlx/backend/common/threefry.cpp b/mlx/backend/common/threefry.cpp
index 391648ce0..b4905acbe 100644
--- a/mlx/backend/common/threefry.cpp
+++ b/mlx/backend/common/threefry.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "mlx/backend/common/threefry.h"
 
 namespace mlx::core::random {
diff --git a/mlx/backend/common/threefry.h b/mlx/backend/common/threefry.h
index efcb5dcfd..0fc485fcb 100644
--- a/mlx/backend/common/threefry.h
+++ b/mlx/backend/common/threefry.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <cstdint>
diff --git a/mlx/backend/common/unary.h b/mlx/backend/common/unary.h
index 271215b53..dbbaf6993 100644
--- a/mlx/backend/common/unary.h
+++ b/mlx/backend/common/unary.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/allocator.h"
diff --git a/mlx/backend/common/utils.h b/mlx/backend/common/utils.h
index 1cc7e9f56..e170f5dfa 100644
--- a/mlx/backend/common/utils.h
+++ b/mlx/backend/common/utils.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <vector>
diff --git a/mlx/backend/metal/allocator.cpp b/mlx/backend/metal/allocator.cpp
index 40ec85a13..a55690947 100644
--- a/mlx/backend/metal/allocator.cpp
+++ b/mlx/backend/metal/allocator.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "mlx/backend/metal/allocator.h"
 #include "mlx/backend/metal/metal.h"
 
diff --git a/mlx/backend/metal/allocator.h b/mlx/backend/metal/allocator.h
index 92ede1757..621b78a92 100644
--- a/mlx/backend/metal/allocator.h
+++ b/mlx/backend/metal/allocator.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <map>
diff --git a/mlx/backend/metal/conv.cpp b/mlx/backend/metal/conv.cpp
index e13297174..e25599caf 100644
--- a/mlx/backend/metal/conv.cpp
+++ b/mlx/backend/metal/conv.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cassert>
 #include <iostream>
diff --git a/mlx/backend/metal/copy.cpp b/mlx/backend/metal/copy.cpp
index aadb98e9a..c14c1022e 100644
--- a/mlx/backend/metal/copy.cpp
+++ b/mlx/backend/metal/copy.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <sstream>
 
 #include "mlx/backend/metal/copy.h"
diff --git a/mlx/backend/metal/copy.h b/mlx/backend/metal/copy.h
index 02d294a84..c08ae2cf7 100644
--- a/mlx/backend/metal/copy.h
+++ b/mlx/backend/metal/copy.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/backend/common/copy.h"
diff --git a/mlx/backend/metal/device.cpp b/mlx/backend/metal/device.cpp
index 6a133329c..24251ca0e 100644
--- a/mlx/backend/metal/device.cpp
+++ b/mlx/backend/metal/device.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <dlfcn.h>
 #include <cstdlib>
 #include <filesystem>
diff --git a/mlx/backend/metal/device.h b/mlx/backend/metal/device.h
index bad001573..62675d430 100644
--- a/mlx/backend/metal/device.h
+++ b/mlx/backend/metal/device.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <Metal/Metal.hpp>
diff --git a/mlx/backend/metal/fft.cpp b/mlx/backend/metal/fft.cpp
index 0cf34e5d8..c0844d374 100644
--- a/mlx/backend/metal/fft.cpp
+++ b/mlx/backend/metal/fft.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "mlx/primitives.h"
 
 namespace mlx::core {
diff --git a/mlx/backend/metal/indexing.cpp b/mlx/backend/metal/indexing.cpp
index ea5b771f2..eb9a8efb6 100644
--- a/mlx/backend/metal/indexing.cpp
+++ b/mlx/backend/metal/indexing.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cassert>
 #include <numeric>
diff --git a/mlx/backend/metal/kernels/arange.metal b/mlx/backend/metal/kernels/arange.metal
index f3afc24f6..d7ba56aae 100644
--- a/mlx/backend/metal/kernels/arange.metal
+++ b/mlx/backend/metal/kernels/arange.metal
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "mlx/backend/metal/kernels/bf16.h"
 
 template <typename T>
diff --git a/mlx/backend/metal/kernels/arg_reduce.metal b/mlx/backend/metal/kernels/arg_reduce.metal
index 8cd13afca..31bcbfa05 100644
--- a/mlx/backend/metal/kernels/arg_reduce.metal
+++ b/mlx/backend/metal/kernels/arg_reduce.metal
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <metal_atomic>
 #include <metal_simdgroup>
 
diff --git a/mlx/backend/metal/kernels/atomic.h b/mlx/backend/metal/kernels/atomic.h
index 43c8cc3ba..1c260d1b6 100644
--- a/mlx/backend/metal/kernels/atomic.h
+++ b/mlx/backend/metal/kernels/atomic.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <metal_atomic>
diff --git a/mlx/backend/metal/kernels/bf16.h b/mlx/backend/metal/kernels/bf16.h
index 618aefd40..71a45e1c5 100644
--- a/mlx/backend/metal/kernels/bf16.h
+++ b/mlx/backend/metal/kernels/bf16.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <metal_stdlib>
diff --git a/mlx/backend/metal/kernels/bf16_math.h b/mlx/backend/metal/kernels/bf16_math.h
index 824ecc9ef..f1fec4336 100644
--- a/mlx/backend/metal/kernels/bf16_math.h
+++ b/mlx/backend/metal/kernels/bf16_math.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/backend/metal/kernels/bf16.h"
diff --git a/mlx/backend/metal/kernels/binary.metal b/mlx/backend/metal/kernels/binary.metal
index 5baab0e96..c22bfde39 100644
--- a/mlx/backend/metal/kernels/binary.metal
+++ b/mlx/backend/metal/kernels/binary.metal
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <metal_integer>
 #include <metal_math>
 
diff --git a/mlx/backend/metal/kernels/complex.h b/mlx/backend/metal/kernels/complex.h
index b4be6d51e..0bfa707f9 100644
--- a/mlx/backend/metal/kernels/complex.h
+++ b/mlx/backend/metal/kernels/complex.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <metal_stdlib>
diff --git a/mlx/backend/metal/kernels/conv.metal b/mlx/backend/metal/kernels/conv.metal
index 65329d19f..6dbe7bc3a 100644
--- a/mlx/backend/metal/kernels/conv.metal
+++ b/mlx/backend/metal/kernels/conv.metal
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <metal_stdlib>
 
 #include "mlx/backend/metal/kernels/conv_params.h"
diff --git a/mlx/backend/metal/kernels/conv_params.h b/mlx/backend/metal/kernels/conv_params.h
index b3748ef46..b216bb976 100644
--- a/mlx/backend/metal/kernels/conv_params.h
+++ b/mlx/backend/metal/kernels/conv_params.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 template <int NDIM>
diff --git a/mlx/backend/metal/kernels/copy.metal b/mlx/backend/metal/kernels/copy.metal
index 07ae22baa..9808cc958 100644
--- a/mlx/backend/metal/kernels/copy.metal
+++ b/mlx/backend/metal/kernels/copy.metal
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "mlx/backend/metal/kernels/bf16.h"
 #include "mlx/backend/metal/kernels/utils.h"
 
diff --git a/mlx/backend/metal/kernels/defines.h b/mlx/backend/metal/kernels/defines.h
index 38102e987..bdd1419f2 100644
--- a/mlx/backend/metal/kernels/defines.h
+++ b/mlx/backend/metal/kernels/defines.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #ifdef __METAL__
diff --git a/mlx/backend/metal/kernels/erf.h b/mlx/backend/metal/kernels/erf.h
index de9cc58c9..0a370d304 100644
--- a/mlx/backend/metal/kernels/erf.h
+++ b/mlx/backend/metal/kernels/erf.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <metal_math>
diff --git a/mlx/backend/metal/kernels/gemm.metal b/mlx/backend/metal/kernels/gemm.metal
index 9f51e6ff8..df150b50d 100644
--- a/mlx/backend/metal/kernels/gemm.metal
+++ b/mlx/backend/metal/kernels/gemm.metal
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "mlx/backend/metal/kernels/bf16.h"
 #include "mlx/backend/metal/kernels/gemm/gemm.h"
 
diff --git a/mlx/backend/metal/kernels/gemm/conv.h b/mlx/backend/metal/kernels/gemm/conv.h
index faf550e34..2c4a7074a 100644
--- a/mlx/backend/metal/kernels/gemm/conv.h
+++ b/mlx/backend/metal/kernels/gemm/conv.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <metal_simdgroup>
diff --git a/mlx/backend/metal/kernels/gemm/gemm.h b/mlx/backend/metal/kernels/gemm/gemm.h
index f6c7b5832..f551947dd 100644
--- a/mlx/backend/metal/kernels/gemm/gemm.h
+++ b/mlx/backend/metal/kernels/gemm/gemm.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <metal_simdgroup>
diff --git a/mlx/backend/metal/kernels/gemv.metal b/mlx/backend/metal/kernels/gemv.metal
index 9b484871b..ea52ab57f 100644
--- a/mlx/backend/metal/kernels/gemv.metal
+++ b/mlx/backend/metal/kernels/gemv.metal
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <metal_stdlib>
 #include <metal_simdgroup>
 
diff --git a/mlx/backend/metal/kernels/indexing.metal b/mlx/backend/metal/kernels/indexing.metal
index fc0d9ee72..daba39a47 100644
--- a/mlx/backend/metal/kernels/indexing.metal
+++ b/mlx/backend/metal/kernels/indexing.metal
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <metal_atomic>
 #include <metal_texture>
 
diff --git a/mlx/backend/metal/kernels/random.metal b/mlx/backend/metal/kernels/random.metal
index d8e4c4208..cc9c5fe30 100644
--- a/mlx/backend/metal/kernels/random.metal
+++ b/mlx/backend/metal/kernels/random.metal
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "mlx/backend/metal/kernels/utils.h"
 
 static constexpr constant uint32_t rotations[2][4] = {
diff --git a/mlx/backend/metal/kernels/reduce.h b/mlx/backend/metal/kernels/reduce.h
index d670f245f..1d2b971b2 100644
--- a/mlx/backend/metal/kernels/reduce.h
+++ b/mlx/backend/metal/kernels/reduce.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <metal_atomic>
diff --git a/mlx/backend/metal/kernels/reduce.metal b/mlx/backend/metal/kernels/reduce.metal
index f69514906..25bf1ee1f 100644
--- a/mlx/backend/metal/kernels/reduce.metal
+++ b/mlx/backend/metal/kernels/reduce.metal
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <metal_atomic>
 #include <metal_simdgroup>
 
diff --git a/mlx/backend/metal/kernels/scan.metal b/mlx/backend/metal/kernels/scan.metal
index ba2368e25..e692449c7 100644
--- a/mlx/backend/metal/kernels/scan.metal
+++ b/mlx/backend/metal/kernels/scan.metal
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <metal_math>
 #include <metal_simdgroup>
 
diff --git a/mlx/backend/metal/kernels/softmax.metal b/mlx/backend/metal/kernels/softmax.metal
index 07a49bc07..0877de075 100644
--- a/mlx/backend/metal/kernels/softmax.metal
+++ b/mlx/backend/metal/kernels/softmax.metal
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <metal_atomic>
 #include <metal_common>
 #include <metal_simdgroup>
diff --git a/mlx/backend/metal/kernels/sort.metal b/mlx/backend/metal/kernels/sort.metal
index adafe6e13..7e2c0442a 100644
--- a/mlx/backend/metal/kernels/sort.metal
+++ b/mlx/backend/metal/kernels/sort.metal
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <metal_stdlib>
 
 #include "mlx/backend/metal/kernels/bf16.h"
diff --git a/mlx/backend/metal/kernels/unary.metal b/mlx/backend/metal/kernels/unary.metal
index 213e440d4..bfda9938c 100644
--- a/mlx/backend/metal/kernels/unary.metal
+++ b/mlx/backend/metal/kernels/unary.metal
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <metal_integer>
 #include <metal_math>
 
diff --git a/mlx/backend/metal/kernels/utils.h b/mlx/backend/metal/kernels/utils.h
index 22b4a271b..72cdd8b20 100644
--- a/mlx/backend/metal/kernels/utils.h
+++ b/mlx/backend/metal/kernels/utils.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <metal_math>
diff --git a/mlx/backend/metal/matmul.cpp b/mlx/backend/metal/matmul.cpp
index d6fdc34ec..22472c638 100644
--- a/mlx/backend/metal/matmul.cpp
+++ b/mlx/backend/metal/matmul.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cassert>
 #include <numeric>
diff --git a/mlx/backend/metal/matmul.h b/mlx/backend/metal/matmul.h
index 34e86733b..78a0e7f26 100644
--- a/mlx/backend/metal/matmul.h
+++ b/mlx/backend/metal/matmul.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cassert>
 #include <sstream>
diff --git a/mlx/backend/metal/metal.cpp b/mlx/backend/metal/metal.cpp
index d936ff3ed..f63ad55a3 100644
--- a/mlx/backend/metal/metal.cpp
+++ b/mlx/backend/metal/metal.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cstdlib>
 #include <future>
 #include <memory>
diff --git a/mlx/backend/metal/metal.h b/mlx/backend/metal/metal.h
index 8195080f6..f1f7ede44 100644
--- a/mlx/backend/metal/metal.h
+++ b/mlx/backend/metal/metal.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <future>
diff --git a/mlx/backend/metal/mps/gemm.h b/mlx/backend/metal/mps/gemm.h
index 620164469..c93df68d8 100644
--- a/mlx/backend/metal/mps/gemm.h
+++ b/mlx/backend/metal/mps/gemm.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <Metal/Metal.hpp>
diff --git a/mlx/backend/metal/primitives.cpp b/mlx/backend/metal/primitives.cpp
index 48ceb0548..7860d7de2 100644
--- a/mlx/backend/metal/primitives.cpp
+++ b/mlx/backend/metal/primitives.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cassert>
 #include <numeric>
diff --git a/mlx/backend/metal/reduce.cpp b/mlx/backend/metal/reduce.cpp
index 84a368c53..532f18353 100644
--- a/mlx/backend/metal/reduce.cpp
+++ b/mlx/backend/metal/reduce.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cassert>
 #include <sstream>
diff --git a/mlx/backend/metal/scan.cpp b/mlx/backend/metal/scan.cpp
index 9c91f89e9..92f1007e1 100644
--- a/mlx/backend/metal/scan.cpp
+++ b/mlx/backend/metal/scan.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 #include <sstream>
 
diff --git a/mlx/backend/metal/softmax.cpp b/mlx/backend/metal/softmax.cpp
index 7f7b51b92..33ec8014c 100644
--- a/mlx/backend/metal/softmax.cpp
+++ b/mlx/backend/metal/softmax.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 
 #include "mlx/backend/metal/copy.h"
diff --git a/mlx/backend/metal/sort.cpp b/mlx/backend/metal/sort.cpp
index 8cffdd4eb..befbf2d81 100644
--- a/mlx/backend/metal/sort.cpp
+++ b/mlx/backend/metal/sort.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 
 #include "mlx/backend/metal/copy.h"
diff --git a/mlx/backend/metal/utils.h b/mlx/backend/metal/utils.h
index 0165761e8..6fa08e42a 100644
--- a/mlx/backend/metal/utils.h
+++ b/mlx/backend/metal/utils.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/array.h"
diff --git a/mlx/backend/no_metal/allocator.cpp b/mlx/backend/no_metal/allocator.cpp
index 718f3509b..dcb67d93d 100644
--- a/mlx/backend/no_metal/allocator.cpp
+++ b/mlx/backend/no_metal/allocator.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 
 #include "mlx/allocator.h"
 
diff --git a/mlx/backend/no_metal/metal.cpp b/mlx/backend/no_metal/metal.cpp
index ec4518887..accfc4c8a 100644
--- a/mlx/backend/no_metal/metal.cpp
+++ b/mlx/backend/no_metal/metal.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <stdexcept>
 
 #include "mlx/backend/metal/metal.h"
diff --git a/mlx/backend/no_metal/primitives.cpp b/mlx/backend/no_metal/primitives.cpp
index 7a42e3ff6..3c327c677 100644
--- a/mlx/backend/no_metal/primitives.cpp
+++ b/mlx/backend/no_metal/primitives.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 
 #include "mlx/primitives.h"
 
diff --git a/mlx/device.cpp b/mlx/device.cpp
index 352b0284c..e635782e2 100644
--- a/mlx/device.cpp
+++ b/mlx/device.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "mlx/device.h"
 #include "mlx/backend/metal/metal.h"
 
diff --git a/mlx/device.h b/mlx/device.h
index 66035ddca..e11edf793 100644
--- a/mlx/device.h
+++ b/mlx/device.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 namespace mlx::core {
diff --git a/mlx/dtype.cpp b/mlx/dtype.cpp
index 23749e9a6..99b75a028 100644
--- a/mlx/dtype.cpp
+++ b/mlx/dtype.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cstdint>
 #include <sstream>
 #include <vector>
diff --git a/mlx/dtype.h b/mlx/dtype.h
index 6a5f1179b..cdb8aa591 100644
--- a/mlx/dtype.h
+++ b/mlx/dtype.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <complex>
diff --git a/mlx/fft.cpp b/mlx/fft.cpp
index 96dc76bab..6cb33048d 100644
--- a/mlx/fft.cpp
+++ b/mlx/fft.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <numeric>
 #include <set>
 
diff --git a/mlx/fft.h b/mlx/fft.h
index 62a34f4e9..dbcc777fe 100644
--- a/mlx/fft.h
+++ b/mlx/fft.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <variant>
diff --git a/mlx/graph_utils.cpp b/mlx/graph_utils.cpp
index 59bd37367..7c83fdeac 100644
--- a/mlx/graph_utils.cpp
+++ b/mlx/graph_utils.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <functional>
 #include <optional>
 #include <sstream>
diff --git a/mlx/graph_utils.h b/mlx/graph_utils.h
index e0696e5ef..3bd373bec 100644
--- a/mlx/graph_utils.h
+++ b/mlx/graph_utils.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/array.h"
diff --git a/mlx/load.cpp b/mlx/load.cpp
index 1d3b571ed..8106448a4 100644
--- a/mlx/load.cpp
+++ b/mlx/load.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cstring>
 #include <fstream>
diff --git a/mlx/load.h b/mlx/load.h
index 8b433ac22..1d193392a 100644
--- a/mlx/load.h
+++ b/mlx/load.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <fstream>
diff --git a/mlx/mlx.h b/mlx/mlx.h
index 68b6c182c..102d2dde9 100644
--- a/mlx/mlx.h
+++ b/mlx/mlx.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/array.h"
diff --git a/mlx/ops.cpp b/mlx/ops.cpp
index c851a44ac..c279b76f6 100644
--- a/mlx/ops.cpp
+++ b/mlx/ops.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cmath>
 #include <numeric>
 #include <set>
diff --git a/mlx/ops.h b/mlx/ops.h
index 50666fa25..552226b89 100644
--- a/mlx/ops.h
+++ b/mlx/ops.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <variant>
diff --git a/mlx/primitives.cpp b/mlx/primitives.cpp
index 8a2ade91e..4c07e14e0 100644
--- a/mlx/primitives.cpp
+++ b/mlx/primitives.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cassert>
 #include <cmath>
diff --git a/mlx/primitives.h b/mlx/primitives.h
index 44159f86d..3497bc262 100644
--- a/mlx/primitives.h
+++ b/mlx/primitives.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "array.h"
diff --git a/mlx/random.cpp b/mlx/random.cpp
index 41ce2b5a6..e976c2b89 100644
--- a/mlx/random.cpp
+++ b/mlx/random.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cmath>
 #include <sstream>
 
diff --git a/mlx/random.h b/mlx/random.h
index 25dc9f924..e684464bc 100644
--- a/mlx/random.h
+++ b/mlx/random.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <optional>
diff --git a/mlx/scheduler.cpp b/mlx/scheduler.cpp
index cb2842a59..af66af06e 100644
--- a/mlx/scheduler.cpp
+++ b/mlx/scheduler.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "mlx/scheduler.h"
 #include "mlx/backend/metal/metal.h"
 
diff --git a/mlx/scheduler.h b/mlx/scheduler.h
index ba615d37e..6506b20ab 100644
--- a/mlx/scheduler.h
+++ b/mlx/scheduler.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <atomic>
diff --git a/mlx/stream.h b/mlx/stream.h
index 000185f0f..d7b4268fd 100644
--- a/mlx/stream.h
+++ b/mlx/stream.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/device.h"
diff --git a/mlx/transforms.cpp b/mlx/transforms.cpp
index 89e0c42a9..174d5e374 100644
--- a/mlx/transforms.cpp
+++ b/mlx/transforms.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <future>
 #include <map>
diff --git a/mlx/transforms.h b/mlx/transforms.h
index 88296521e..ff47638bf 100644
--- a/mlx/transforms.h
+++ b/mlx/transforms.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "array.h"
diff --git a/mlx/transforms_impl.h b/mlx/transforms_impl.h
index e97d41d73..0fa8f06a2 100644
--- a/mlx/transforms_impl.h
+++ b/mlx/transforms_impl.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 
 namespace mlx::core::detail {
 
diff --git a/mlx/types/bf16.h b/mlx/types/bf16.h
index 102aea91e..595194174 100644
--- a/mlx/types/bf16.h
+++ b/mlx/types/bf16.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <algorithm>
diff --git a/mlx/types/complex.h b/mlx/types/complex.h
index b7533efba..55cbe447a 100644
--- a/mlx/types/complex.h
+++ b/mlx/types/complex.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 #include <complex>
 #include "mlx/types/half_types.h"
diff --git a/mlx/types/fp16.h b/mlx/types/fp16.h
index e4f94c994..58e1bc088 100644
--- a/mlx/types/fp16.h
+++ b/mlx/types/fp16.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <algorithm>
diff --git a/mlx/types/half_types.h b/mlx/types/half_types.h
index 47ea3d08e..430279565 100644
--- a/mlx/types/half_types.h
+++ b/mlx/types/half_types.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 #ifdef __ARM_FEATURE_FP16_SCALAR_ARITHMETIC
 
diff --git a/mlx/utils.cpp b/mlx/utils.cpp
index 027e104e9..5f38cd06f 100644
--- a/mlx/utils.cpp
+++ b/mlx/utils.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <sstream>
 #include <vector>
 
diff --git a/mlx/utils.h b/mlx/utils.h
index b004ae8ba..f21e55fce 100644
--- a/mlx/utils.h
+++ b/mlx/utils.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "array.h"
diff --git a/python/mlx/_reprlib_fix.py b/python/mlx/_reprlib_fix.py
index 51111e5dd..02bf661dc 100644
--- a/python/mlx/_reprlib_fix.py
+++ b/python/mlx/_reprlib_fix.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import array
 import reprlib
 
diff --git a/python/mlx/extension.py b/python/mlx/extension.py
index 8dd7a6a72..2924ef0c6 100644
--- a/python/mlx/extension.py
+++ b/python/mlx/extension.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import os
 import re
 import subprocess
diff --git a/python/mlx/nn/__init__.py b/python/mlx/nn/__init__.py
index 4991254f8..2af4f3a51 100644
--- a/python/mlx/nn/__init__.py
+++ b/python/mlx/nn/__init__.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 from mlx.nn.layers import *
 from mlx.nn import losses
 from mlx.nn.utils import value_and_grad
diff --git a/python/mlx/nn/layers/__init__.py b/python/mlx/nn/layers/__init__.py
index 0436441af..30a185044 100644
--- a/python/mlx/nn/layers/__init__.py
+++ b/python/mlx/nn/layers/__init__.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 from mlx.nn.layers.base import Module
 from mlx.nn.layers.activations import (
     GELU,
diff --git a/python/mlx/nn/layers/activations.py b/python/mlx/nn/layers/activations.py
index ef1cde035..dbe5f4adf 100644
--- a/python/mlx/nn/layers/activations.py
+++ b/python/mlx/nn/layers/activations.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import math
 
 import mlx.core as mx
diff --git a/python/mlx/nn/layers/base.py b/python/mlx/nn/layers/base.py
index 3a3cd3d26..991b13813 100644
--- a/python/mlx/nn/layers/base.py
+++ b/python/mlx/nn/layers/base.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import textwrap
 from typing import Any, Callable, List, Union, Optional
 
diff --git a/python/mlx/nn/layers/containers.py b/python/mlx/nn/layers/containers.py
index 8403be069..c57fbb992 100644
--- a/python/mlx/nn/layers/containers.py
+++ b/python/mlx/nn/layers/containers.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 from mlx.nn.layers.base import Module
 
 
diff --git a/python/mlx/nn/layers/convolution.py b/python/mlx/nn/layers/convolution.py
index 1e7ac1c3d..c9b6ce363 100644
--- a/python/mlx/nn/layers/convolution.py
+++ b/python/mlx/nn/layers/convolution.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import math
 from typing import Union
 
diff --git a/python/mlx/nn/layers/dropout.py b/python/mlx/nn/layers/dropout.py
index e6e889387..3193cdbd7 100644
--- a/python/mlx/nn/layers/dropout.py
+++ b/python/mlx/nn/layers/dropout.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import mlx.core as mx
 from mlx.nn.layers.base import Module
 
diff --git a/python/mlx/nn/layers/embedding.py b/python/mlx/nn/layers/embedding.py
index d6e9b976a..c62b1206f 100644
--- a/python/mlx/nn/layers/embedding.py
+++ b/python/mlx/nn/layers/embedding.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import math
 
 import mlx.core as mx
diff --git a/python/mlx/nn/layers/linear.py b/python/mlx/nn/layers/linear.py
index aa48688d6..d4a503384 100644
--- a/python/mlx/nn/layers/linear.py
+++ b/python/mlx/nn/layers/linear.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import math
 
 import mlx.core as mx
diff --git a/python/mlx/nn/layers/normalization.py b/python/mlx/nn/layers/normalization.py
index 6d0dd4b3a..348a8c545 100644
--- a/python/mlx/nn/layers/normalization.py
+++ b/python/mlx/nn/layers/normalization.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import mlx.core as mx
 from mlx.nn.layers.base import Module
 
diff --git a/python/mlx/nn/layers/positional_encoding.py b/python/mlx/nn/layers/positional_encoding.py
index 9ccc2b692..5c3171fde 100644
--- a/python/mlx/nn/layers/positional_encoding.py
+++ b/python/mlx/nn/layers/positional_encoding.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import math
 from typing import Optional
 
diff --git a/python/mlx/nn/layers/transformer.py b/python/mlx/nn/layers/transformer.py
index 93359a4c7..d2c224cbf 100644
--- a/python/mlx/nn/layers/transformer.py
+++ b/python/mlx/nn/layers/transformer.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import math
 from typing import Optional
 
diff --git a/python/mlx/nn/losses.py b/python/mlx/nn/losses.py
index 800242c70..dfa806c34 100644
--- a/python/mlx/nn/losses.py
+++ b/python/mlx/nn/losses.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import mlx.core as mx
 
 
diff --git a/python/mlx/nn/utils.py b/python/mlx/nn/utils.py
index c1023619f..8ab80cb00 100644
--- a/python/mlx/nn/utils.py
+++ b/python/mlx/nn/utils.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 from typing import Callable
 
 import mlx.core as mx
diff --git a/python/mlx/optimizers.py b/python/mlx/optimizers.py
index 63ccf03d8..4fb48935b 100644
--- a/python/mlx/optimizers.py
+++ b/python/mlx/optimizers.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import math
 from typing import List
 
diff --git a/python/mlx/utils.py b/python/mlx/utils.py
index 47842d83d..d70b8ce57 100644
--- a/python/mlx/utils.py
+++ b/python/mlx/utils.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 def tree_map(fn, tree, *rest):
     """Applies ``fn`` to the leaves of the python tree ``tree`` and
     returns a new collection with the results.
diff --git a/python/src/array.cpp b/python/src/array.cpp
index 5e34bfdbd..ea580c78f 100644
--- a/python/src/array.cpp
+++ b/python/src/array.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cstdint>
 #include <cstring>
 #include <sstream>
diff --git a/python/src/device.cpp b/python/src/device.cpp
index 45ae44aec..50438c139 100644
--- a/python/src/device.cpp
+++ b/python/src/device.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <sstream>
 
 #include <pybind11/pybind11.h>
diff --git a/python/src/fft.cpp b/python/src/fft.cpp
index d20e2b2fe..42ad37633 100644
--- a/python/src/fft.cpp
+++ b/python/src/fft.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <pybind11/pybind11.h>
 #include <pybind11/stl.h>
 
diff --git a/python/src/indexing.cpp b/python/src/indexing.cpp
index 968a53747..788e2fd5f 100644
--- a/python/src/indexing.cpp
+++ b/python/src/indexing.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <numeric>
 #include <sstream>
 
diff --git a/python/src/indexing.h b/python/src/indexing.h
index cb5d3106c..0eaaac93d 100644
--- a/python/src/indexing.h
+++ b/python/src/indexing.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <pybind11/pybind11.h>
diff --git a/python/src/load.cpp b/python/src/load.cpp
index d3f1f9250..b0327e125 100644
--- a/python/src/load.cpp
+++ b/python/src/load.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <pybind11/pybind11.h>
 #include <pybind11/stl.h>
 
diff --git a/python/src/load.h b/python/src/load.h
index f7f1a4148..bbabef2c9 100644
--- a/python/src/load.h
+++ b/python/src/load.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <pybind11/pybind11.h>
diff --git a/python/src/metal.cpp b/python/src/metal.cpp
index 75e0437bf..41b90d974 100644
--- a/python/src/metal.cpp
+++ b/python/src/metal.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <pybind11/pybind11.h>
 
 #include "mlx/backend/metal/metal.h"
diff --git a/python/src/mlx.cpp b/python/src/mlx.cpp
index 0e9aeba65..4a1d24aef 100644
--- a/python/src/mlx.cpp
+++ b/python/src/mlx.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <pybind11/pybind11.h>
 
 #define STRINGIFY(x) #x
diff --git a/python/src/ops.cpp b/python/src/ops.cpp
index ca33cabd2..e32c01bcf 100644
--- a/python/src/ops.cpp
+++ b/python/src/ops.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <numeric>
 #include <ostream>
 #include <variant>
diff --git a/python/src/random.cpp b/python/src/random.cpp
index f38fc43d9..943370db5 100644
--- a/python/src/random.cpp
+++ b/python/src/random.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <pybind11/pybind11.h>
 #include <pybind11/stl.h>
 
diff --git a/python/src/stream.cpp b/python/src/stream.cpp
index f7a73463f..7b1b2f55d 100644
--- a/python/src/stream.cpp
+++ b/python/src/stream.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <sstream>
 
 #include <pybind11/pybind11.h>
diff --git a/python/src/transforms.cpp b/python/src/transforms.cpp
index b257ab40f..a592b4458 100644
--- a/python/src/transforms.cpp
+++ b/python/src/transforms.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <pybind11/functional.h>
 #include <pybind11/pybind11.h>
 #include <pybind11/stl.h>
diff --git a/python/src/utils.h b/python/src/utils.h
index bc06fb504..142f9a985 100644
--- a/python/src/utils.h
+++ b/python/src/utils.h
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 #include <numeric>
 #include <variant>
diff --git a/python/tests/mlx_tests.py b/python/tests/mlx_tests.py
index 6c19f6eb8..7bbadc9cb 100644
--- a/python/tests/mlx_tests.py
+++ b/python/tests/mlx_tests.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import os
 import unittest
 
diff --git a/python/tests/test_array.py b/python/tests/test_array.py
index e7edb6c20..16579742a 100644
--- a/python/tests/test_array.py
+++ b/python/tests/test_array.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import operator
 import unittest
 from itertools import permutations
diff --git a/python/tests/test_autograd.py b/python/tests/test_autograd.py
index 96e22dd6b..5cead2f33 100644
--- a/python/tests/test_autograd.py
+++ b/python/tests/test_autograd.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 
 import mlx.core as mx
diff --git a/python/tests/test_bf16.py b/python/tests/test_bf16.py
index e73937d79..27a4e5623 100644
--- a/python/tests/test_bf16.py
+++ b/python/tests/test_bf16.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 from itertools import permutations
 
diff --git a/python/tests/test_blas.py b/python/tests/test_blas.py
index 3a81e10c7..4b310ed33 100644
--- a/python/tests/test_blas.py
+++ b/python/tests/test_blas.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 from itertools import permutations
 
diff --git a/python/tests/test_conv.py b/python/tests/test_conv.py
index c4df60357..13b51e711 100644
--- a/python/tests/test_conv.py
+++ b/python/tests/test_conv.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 from itertools import permutations
 
diff --git a/python/tests/test_device.py b/python/tests/test_device.py
index b3abec01f..fff368b9a 100644
--- a/python/tests/test_device.py
+++ b/python/tests/test_device.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 
 import mlx.core as mx
diff --git a/python/tests/test_eval.py b/python/tests/test_eval.py
index e4d48a877..88149729c 100644
--- a/python/tests/test_eval.py
+++ b/python/tests/test_eval.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 from functools import partial
 
 import unittest
diff --git a/python/tests/test_fft.py b/python/tests/test_fft.py
index 6af90e1c5..ce273bb22 100644
--- a/python/tests/test_fft.py
+++ b/python/tests/test_fft.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 
 import itertools
diff --git a/python/tests/test_load.py b/python/tests/test_load.py
index ebaddf985..6f345fa99 100644
--- a/python/tests/test_load.py
+++ b/python/tests/test_load.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 import os
 import mlx.core as mx
diff --git a/python/tests/test_nn.py b/python/tests/test_nn.py
index cc664cbb0..77eb6bf79 100644
--- a/python/tests/test_nn.py
+++ b/python/tests/test_nn.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 
 import mlx.core as mx
diff --git a/python/tests/test_ops.py b/python/tests/test_ops.py
index 9885a7e70..7def06c29 100644
--- a/python/tests/test_ops.py
+++ b/python/tests/test_ops.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 from itertools import permutations
 
diff --git a/python/tests/test_optimizers.py b/python/tests/test_optimizers.py
index 7cfc523f3..3d4f4518f 100644
--- a/python/tests/test_optimizers.py
+++ b/python/tests/test_optimizers.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 
 import mlx.core as mx
diff --git a/python/tests/test_random.py b/python/tests/test_random.py
index 209d6d9e5..e850e4235 100644
--- a/python/tests/test_random.py
+++ b/python/tests/test_random.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 
 import mlx.core as mx
diff --git a/python/tests/test_reduce.py b/python/tests/test_reduce.py
index b75f8a0ba..da6207325 100644
--- a/python/tests/test_reduce.py
+++ b/python/tests/test_reduce.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 from itertools import permutations, combinations
 
diff --git a/python/tests/test_tree.py b/python/tests/test_tree.py
index fe6d8cdb7..f5003c4a7 100644
--- a/python/tests/test_tree.py
+++ b/python/tests/test_tree.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 
 import mlx.core as mx
diff --git a/python/tests/test_vmap.py b/python/tests/test_vmap.py
index 151fbc3ca..50f7e6ef6 100644
--- a/python/tests/test_vmap.py
+++ b/python/tests/test_vmap.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 
 import mlx.core as mx
diff --git a/setup.py b/setup.py
index c4d3569cb..c5e6c1de2 100644
--- a/setup.py
+++ b/setup.py
@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import os
 import re
 import subprocess
diff --git a/tests/allocator_tests.cpp b/tests/allocator_tests.cpp
index 70199eb46..0c1f30008 100644
--- a/tests/allocator_tests.cpp
+++ b/tests/allocator_tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <stdexcept>
 
 #include "doctest/doctest.h"
diff --git a/tests/arg_reduce_tests.cpp b/tests/arg_reduce_tests.cpp
index 2bfd8968f..7e3750e7b 100644
--- a/tests/arg_reduce_tests.cpp
+++ b/tests/arg_reduce_tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <iostream>
 
 #include "doctest/doctest.h"
diff --git a/tests/array_tests.cpp b/tests/array_tests.cpp
index c435b1db4..334b86115 100644
--- a/tests/array_tests.cpp
+++ b/tests/array_tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <climits>
 
 #include "doctest/doctest.h"
diff --git a/tests/autograd_tests.cpp b/tests/autograd_tests.cpp
index 7c7245cd7..85dad9160 100644
--- a/tests/autograd_tests.cpp
+++ b/tests/autograd_tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cmath>
 #include <numeric>
diff --git a/tests/blas_tests.cpp b/tests/blas_tests.cpp
index 256a876a3..f700e8c8a 100644
--- a/tests/blas_tests.cpp
+++ b/tests/blas_tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <numeric>
 
 #include "doctest/doctest.h"
diff --git a/tests/creations_tests.cpp b/tests/creations_tests.cpp
index 8d03ff1f9..9d5f787c5 100644
--- a/tests/creations_tests.cpp
+++ b/tests/creations_tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "doctest/doctest.h"
 
 #include "mlx/mlx.h"
diff --git a/tests/device_tests.cpp b/tests/device_tests.cpp
index dedf64571..87d55d644 100644
--- a/tests/device_tests.cpp
+++ b/tests/device_tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "doctest/doctest.h"
 
 #include <cstdlib>
diff --git a/tests/eval_tests.cpp b/tests/eval_tests.cpp
index 88af76d58..dc2e96bbb 100644
--- a/tests/eval_tests.cpp
+++ b/tests/eval_tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "doctest/doctest.h"
 
 #include "mlx/mlx.h"
diff --git a/tests/fft_tests.cpp b/tests/fft_tests.cpp
index c6940751b..5d960ef10 100644
--- a/tests/fft_tests.cpp
+++ b/tests/fft_tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "doctest/doctest.h"
 
 #include "mlx/mlx.h"
diff --git a/tests/graph_optimize_tests.cpp b/tests/graph_optimize_tests.cpp
index 669bb08ec..8c2da162c 100644
--- a/tests/graph_optimize_tests.cpp
+++ b/tests/graph_optimize_tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "doctest/doctest.h"
 
 #include "mlx/mlx.h"
diff --git a/tests/load_tests.cpp b/tests/load_tests.cpp
index 45f36e473..f2489ca72 100644
--- a/tests/load_tests.cpp
+++ b/tests/load_tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <filesystem>
 #include <stdexcept>
 #include <vector>
diff --git a/tests/metal_tests.cpp b/tests/metal_tests.cpp
index 76f62dce0..adeeb742a 100644
--- a/tests/metal_tests.cpp
+++ b/tests/metal_tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <array>
 #include "doctest/doctest.h"
 
diff --git a/tests/ops_tests.cpp b/tests/ops_tests.cpp
index a9a74b829..5dcf8658d 100644
--- a/tests/ops_tests.cpp
+++ b/tests/ops_tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cmath>
 #include <numeric>
 
diff --git a/tests/random_tests.cpp b/tests/random_tests.cpp
index 92ed802de..c878205e0 100644
--- a/tests/random_tests.cpp
+++ b/tests/random_tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <numeric>
 
 #include "doctest/doctest.h"
diff --git a/tests/scheduler_tests.cpp b/tests/scheduler_tests.cpp
index 781b2aa70..d30d01cfc 100644
--- a/tests/scheduler_tests.cpp
+++ b/tests/scheduler_tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "doctest/doctest.h"
 
 #include "mlx/mlx.h"
diff --git a/tests/tests.cpp b/tests/tests.cpp
index 0ad8b6018..edaf49c27 100644
--- a/tests/tests.cpp
+++ b/tests/tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #define DOCTEST_CONFIG_IMPLEMENT
 #include "doctest/doctest.h"
 
diff --git a/tests/utils_tests.cpp b/tests/utils_tests.cpp
index 867a2cf58..35e28adc3 100644
--- a/tests/utils_tests.cpp
+++ b/tests/utils_tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 
 #include "doctest/doctest.h"
 
diff --git a/tests/vmap_tests.cpp b/tests/vmap_tests.cpp
index 6ccc25d47..41175b0f5 100644
--- a/tests/vmap_tests.cpp
+++ b/tests/vmap_tests.cpp
@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "doctest/doctest.h"
 
 #include "mlx/mlx.h"