diff --git a/jmp/BUILD b/jmp/BUILD
index 000cac2..4d8a770 100644
--- a/jmp/BUILD
+++ b/jmp/BUILD
@@ -1,7 +1,16 @@
 load("//third_party/bazel_rules/rules_python/python:py_library.bzl", "py_library")
+load("//tools/build_defs/license:license.bzl", "license")
 
 # Description: JMP is a JAX Mixed Precision library.
-package(default_visibility = ["//visibility:private"])
+package(
+    default_applicable_licenses = [":license"],
+    default_visibility = ["//visibility:private"],
+)
+
+license(
+    name = "license",
+    package_name = "jmp",
+)
 
 licenses(["notice"])
 
diff --git a/jmp/_src/BUILD b/jmp/_src/BUILD
index abf2c3b..f6ce549 100644
--- a/jmp/_src/BUILD
+++ b/jmp/_src/BUILD
@@ -1,6 +1,9 @@
 load("//jmp/_src:build_defs.bzl", "jmp_py_library", "jmp_py_test")
 
-package(default_visibility = ["//jmp:__subpackages__"])
+package(
+    default_applicable_licenses = ["//jmp:license"],
+    default_visibility = ["//jmp:__subpackages__"],
+)
 
 licenses(["notice"])