diff --git a/axonn/intra_layer/__init__.py b/axonn/intra_layer/__init__.py
index 683754d..7c3b31f 100644
--- a/axonn/intra_layer/__init__.py
+++ b/axonn/intra_layer/__init__.py
@@ -1,3 +1,8 @@
+# Copyright 2021 Parallel Software and Systems Group, University of Maryland.
+# See the top-level LICENSE file for details.
+#
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
 from contextlib import contextmanager
 from .fully_connected import Linear  # noqa: F401
 from .conv import Conv2d  # noqa: F401
diff --git a/axonn/intra_layer/embedding.py b/axonn/intra_layer/embedding.py
index fcd5e5e..b2f2d42 100644
--- a/axonn/intra_layer/embedding.py
+++ b/axonn/intra_layer/embedding.py
@@ -1,3 +1,8 @@
+# Copyright 2021 Parallel Software and Systems Group, University of Maryland.
+# See the top-level LICENSE file for details.
+#
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
 import torch.distributed as dist
 import torch
 import torch.nn.functional as F
diff --git a/axonn/intra_layer/gradient_normalization.py b/axonn/intra_layer/gradient_normalization.py
index 71d880f..7989b0a 100644
--- a/axonn/intra_layer/gradient_normalization.py
+++ b/axonn/intra_layer/gradient_normalization.py
@@ -1,3 +1,8 @@
+# Copyright 2021 Parallel Software and Systems Group, University of Maryland.
+# See the top-level LICENSE file for details.
+#
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
 import torch
 
 # for backwards compatibility with pytorch 1.13
diff --git a/axonn/tests/test_intra_layer_conv.py b/axonn/tests/test_intra_layer_conv.py
index 00361a2..787babb 100644
--- a/axonn/tests/test_intra_layer_conv.py
+++ b/axonn/tests/test_intra_layer_conv.py
@@ -1,3 +1,8 @@
+# Copyright 2021 Parallel Software and Systems Group, University of Maryland.
+# See the top-level LICENSE file for details.
+#
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
 import torch
 import pytest
 from mpi4py import MPI  # noqa: F401
diff --git a/axonn/tests/test_intra_layer_emb.py b/axonn/tests/test_intra_layer_emb.py
index c354617..1fee5fc 100644
--- a/axonn/tests/test_intra_layer_emb.py
+++ b/axonn/tests/test_intra_layer_emb.py
@@ -1,3 +1,8 @@
+# Copyright 2021 Parallel Software and Systems Group, University of Maryland.
+# See the top-level LICENSE file for details.
+#
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
 import torch
 import pytest
 from axonn import axonn as ax
diff --git a/axonn/tests/test_intra_layer_fc.py b/axonn/tests/test_intra_layer_fc.py
index 32d8e7e..507a026 100644
--- a/axonn/tests/test_intra_layer_fc.py
+++ b/axonn/tests/test_intra_layer_fc.py
@@ -1,3 +1,8 @@
+# Copyright 2021 Parallel Software and Systems Group, University of Maryland.
+# See the top-level LICENSE file for details.
+#
+# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+
 import torch
 import pytest
 from axonn import axonn as ax