copyright + ack

python/mlx/_reprlib_fix.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import array
 import reprlib
 

python/mlx/extension.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import os
 import re
 import subprocess

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

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,

python/mlx/nn/layers/activations.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import math
 
 import mlx.core as mx

python/mlx/nn/layers/base.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import textwrap
 from typing import Any, Callable, List, Union, Optional
 

python/mlx/nn/layers/containers.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 from mlx.nn.layers.base import Module
 
 

python/mlx/nn/layers/convolution.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import math
 from typing import Union
 

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
 

python/mlx/nn/layers/embedding.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import math
 
 import mlx.core as mx

python/mlx/nn/layers/linear.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import math
 
 import mlx.core as mx

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
 

python/mlx/nn/layers/positional_encoding.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import math
 from typing import Optional
 

python/mlx/nn/layers/transformer.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import math
 from typing import Optional
 

python/mlx/nn/losses.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import mlx.core as mx
 
 

python/mlx/nn/utils.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 from typing import Callable
 
 import mlx.core as mx

python/mlx/optimizers.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import math
 from typing import List
 

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.

python/src/array.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cstdint>
 #include <cstring>
 #include <sstream>

python/src/device.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <sstream>
 
 #include <pybind11/pybind11.h>

python/src/fft.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <pybind11/pybind11.h>
 #include <pybind11/stl.h>
 

python/src/indexing.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <numeric>
 #include <sstream>
 

python/src/indexing.h

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <pybind11/pybind11.h>

python/src/load.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <pybind11/pybind11.h>
 #include <pybind11/stl.h>
 

python/src/load.h

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include <pybind11/pybind11.h>

python/src/metal.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <pybind11/pybind11.h>
 
 #include "mlx/backend/metal/metal.h"

python/src/mlx.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <pybind11/pybind11.h>
 
 #define STRINGIFY(x) #x

python/src/ops.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <numeric>
 #include <ostream>
 #include <variant>

python/src/random.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <pybind11/pybind11.h>
 #include <pybind11/stl.h>
 

python/src/stream.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <sstream>
 
 #include <pybind11/pybind11.h>

python/src/transforms.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <pybind11/functional.h>
 #include <pybind11/pybind11.h>
 #include <pybind11/stl.h>

python/src/utils.h

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 #include <numeric>
 #include <variant>

python/tests/mlx_tests.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import os
 import unittest
 

python/tests/test_array.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import operator
 import unittest
 from itertools import permutations

python/tests/test_autograd.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 
 import mlx.core as mx

python/tests/test_bf16.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 from itertools import permutations
 

python/tests/test_blas.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 from itertools import permutations
 

python/tests/test_conv.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 from itertools import permutations
 

python/tests/test_device.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 
 import mlx.core as mx

python/tests/test_eval.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 from functools import partial
 
 import unittest

python/tests/test_fft.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 
 import itertools

python/tests/test_load.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 import os
 import mlx.core as mx

python/tests/test_nn.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 
 import mlx.core as mx

python/tests/test_ops.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 from itertools import permutations
 

python/tests/test_optimizers.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 
 import mlx.core as mx

python/tests/test_random.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 
 import mlx.core as mx

python/tests/test_reduce.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 from itertools import permutations, combinations
 

python/tests/test_tree.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 
 import mlx.core as mx

python/tests/test_vmap.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import unittest
 
 import mlx.core as mx