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.