diff --git a/BUILD b/BUILD
index 1395e33e0..f42af0049 100644
--- a/BUILD
+++ b/BUILD
@@ -1,5 +1,11 @@
-package(default_visibility = ["//visibility:private"])
+package(default_visibility = ["//visibility:public"])
 
-licenses(["notice"])
+licenses(
+    ["notice"],
+    default_visibility = ["//visibility:public"],
+)
 
-exports_files(["LICENSE"])
+exports_files(
+    ["LICENSE"],
+    default_visibility = ["//visibility:public"],
+)
diff --git a/checkpoint/orbax/checkpoint/_src/path/BUILD b/checkpoint/orbax/checkpoint/_src/path/BUILD
index 5ac980230..2f3982364 100644
--- a/checkpoint/orbax/checkpoint/_src/path/BUILD
+++ b/checkpoint/orbax/checkpoint/_src/path/BUILD
@@ -1,21 +1,17 @@
-load("//devtools/python/blaze:pytype.bzl", "pytype_strict_library", pytype_strict_test = "pytype_strict_contrib_test")
-load("//devtools/python/blaze:strict.bzl", "py_library_providing_imports_info")
-
 package(
-    default_applicable_licenses = ["//third_party/py/orbax/checkpoint:package_license"],
-    default_visibility = [
-        "//third_party/py/orbax/checkpoint:__subpackages__",
-    ],
+    default_applicable_licenses = ["//:package_license"],
+    default_visibility = ["//visibility:public"],
 )
 
 py_library(
     name = "utils",
     srcs = ["utils.py"],
+    default_visibility = ["//visibility:public"],
     srcs_version = "PY3",
 )
 
 py_library(
     name = "path",
     srcs = ["__init__.py"],
-    lib_rule = pytype_strict_library,
+    default_visibility = ["//visibility:public"],
 )
diff --git a/checkpoint/orbax/checkpoint/_src/path/snapshot/BUILD b/checkpoint/orbax/checkpoint/_src/path/snapshot/BUILD
index 54cef2ce6..968dc98b0 100644
--- a/checkpoint/orbax/checkpoint/_src/path/snapshot/BUILD
+++ b/checkpoint/orbax/checkpoint/_src/path/snapshot/BUILD
@@ -1,27 +1,20 @@
-load("//devtools/python/blaze:pytype.bzl", "pytype_strict_library")
-load("//devtools/python/blaze:strict.bzl", "py_strict_test")
-
 package(
-    default_applicable_licenses = ["//third_party/py/orbax/checkpoint:package_license"],
-    default_visibility = [
-        "//learning/deepmind/jax/roc:__subpackages__",
-        "//third_party/py/orbax/checkpoint:__subpackages__",
-    ],
+    default_applicable_licenses = ["//:package_license"],
+    default_visibility = ["//visibility:public"],
 )
 
 py_library(
     name = "snapshot",
     srcs = ["snapshot.py"],
+    default_visibility = ["//visibility:public"],
     srcs_version = "PY3",
-    deps = [
-        "//pyglib:timer",
-        "//third_party/py/orbax/checkpoint/_src/path:utils",
-    ],
+    deps = ["//orbax/checkpoint/orbax/checkpoint/_src/path:utils"],
 )
 
 py_test(
     name = "snapshot_test",
     srcs = ["snapshot_test.py"],
+    default_visibility = ["//visibility:public"],
     srcs_version = "PY3",
     deps = [
         ":snapshot",