copyright + ack

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.

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

benchmarks/cpp/autograd.cpp

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

benchmarks/cpp/compare_devices.cpp

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

benchmarks/cpp/irregular_strides.cpp

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

benchmarks/cpp/single_ops.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "mlx/mlx.h"
 #include "time_utils.h"
 

benchmarks/cpp/time_utils.h

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

benchmarks/numpy/single_ops.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 import numpy as np
 
 from time_utils import time_fn

benchmarks/numpy/time_utils.py

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

benchmarks/python/batch_matmul_bench.py

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

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

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

benchmarks/python/comparative/bench_mlx.py

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

benchmarks/python/comparative/bench_torch.py

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

benchmarks/python/comparative/compare.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 #!/usr/bin/env python
 
 import argparse

benchmarks/python/llama_jax_bench.py

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

benchmarks/python/llama_mlx_bench.py

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

benchmarks/python/llama_torch_bench.py

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

benchmarks/python/single_ops.py

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

benchmarks/python/time_utils.py

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

docs/src/conf.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 # -*- coding: utf-8 -*-
 
 import os

examples/cpp/linear_regression.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <chrono>
 #include <cmath>
 #include <iostream>

examples/cpp/logistic_regression.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <chrono>
 #include <cmath>
 #include <iostream>

examples/cpp/timer.h

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

examples/cpp/tutorial.cpp

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

examples/extensions/axpby/axpby.cpp

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

examples/extensions/axpby/axpby.h

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/ops.h"

examples/extensions/axpby/axpby.metal

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

examples/extensions/bindings.cpp

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

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 *

examples/extensions/setup.py

@@ -1,3 +1,5 @@
+# Copyright © 2023 Apple Inc.
+
 from mlx import extension
 from setuptools import setup
 

examples/python/linear_regression.py

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

examples/python/logistic_regression.py

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

mlx/allocator.cpp

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

mlx/allocator.h

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

mlx/array.cpp

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

mlx/array.h

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

mlx/backend/accelerate/conv.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 
 #include <simd/vector.h>

mlx/backend/accelerate/matmul.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 
 #include <vecLib/BNNS/bnns.h>

mlx/backend/accelerate/primitives.cpp

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

mlx/backend/accelerate/reduce.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 
 #include <simd/vector.h>

mlx/backend/accelerate/softmax.cpp

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

mlx/backend/accelerate/utils.h

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

mlx/backend/common/arange.h

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/allocator.h"

mlx/backend/common/arg_reduce.cpp

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

mlx/backend/common/binary.cpp

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

mlx/backend/common/binary.h

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/allocator.h"

mlx/backend/common/conv.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 
 #ifdef ACCELERATE_NEW_LAPACK

mlx/backend/common/copy.cpp

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

mlx/backend/common/copy.h

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/array.h"

mlx/backend/common/default_primitives.cpp

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

mlx/backend/common/erf.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cmath>
 
 namespace mlx::core {

mlx/backend/common/erf.h

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 
 namespace mlx::core {
 

mlx/backend/common/fft.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <numeric>
 
 #include "mlx/3rdparty/pocketfft.h"

mlx/backend/common/indexing.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cassert>
 #include <cmath>

mlx/backend/common/load.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cassert>
 #include <utility>

mlx/backend/common/primitives.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cassert>
 #include <cmath>

mlx/backend/common/reduce.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <cassert>
 #include <functional>
 #include <limits>

mlx/backend/common/reduce.h

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/backend/common/utils.h"

mlx/backend/common/scan.cpp

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

mlx/backend/common/softmax.cpp

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

mlx/backend/common/sort.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cassert>
 #include <cmath>

mlx/backend/common/threefry.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "mlx/backend/common/threefry.h"
 
 namespace mlx::core::random {

mlx/backend/common/threefry.h

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

mlx/backend/common/unary.h

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/allocator.h"

mlx/backend/common/utils.h

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

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"
 

mlx/backend/metal/allocator.h

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

mlx/backend/metal/conv.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <algorithm>
 #include <cassert>
 #include <iostream>

mlx/backend/metal/copy.cpp

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

mlx/backend/metal/copy.h

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/backend/common/copy.h"

mlx/backend/metal/device.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <dlfcn.h>
 #include <cstdlib>
 #include <filesystem>

mlx/backend/metal/device.h

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

mlx/backend/metal/fft.cpp

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "mlx/primitives.h"
 
 namespace mlx::core {

mlx/backend/metal/indexing.cpp

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

mlx/backend/metal/kernels/arange.metal

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include "mlx/backend/metal/kernels/bf16.h"
 
 template <typename T>

mlx/backend/metal/kernels/arg_reduce.metal

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

mlx/backend/metal/kernels/atomic.h

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

mlx/backend/metal/kernels/bf16.h

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

mlx/backend/metal/kernels/bf16_math.h

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #include "mlx/backend/metal/kernels/bf16.h"

mlx/backend/metal/kernels/binary.metal

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

mlx/backend/metal/kernels/complex.h

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

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"

mlx/backend/metal/kernels/conv_params.h

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 template <int NDIM>

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"
 

mlx/backend/metal/kernels/defines.h

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #pragma once
 
 #ifdef __METAL__

mlx/backend/metal/kernels/erf.h

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

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"
 

mlx/backend/metal/kernels/gemm/conv.h

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

mlx/backend/metal/kernels/gemm/gemm.h

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

mlx/backend/metal/kernels/gemv.metal

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

mlx/backend/metal/kernels/indexing.metal

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

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] = {

mlx/backend/metal/kernels/reduce.h

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

mlx/backend/metal/kernels/reduce.metal

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

mlx/backend/metal/kernels/scan.metal

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

mlx/backend/metal/kernels/softmax.metal

@@ -1,3 +1,5 @@
+// Copyright © 2023 Apple Inc.
+
 #include <metal_atomic>
 #include <metal_common>
 #include <metal_simdgroup>

mlx/backend/metal/kernels/sort.metal

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

mlx/backend/metal/kernels/unary.metal

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

mlx/backend/metal/kernels/utils.h

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