From 5dfab8e2e3f086740bed961f8a18402e5eb4da12 Mon Sep 17 00:00:00 2001
From: Richard Levasseur <rlevasseur@google.com>
Date: Fri, 21 Jul 2023 02:24:15 -0700
Subject: [PATCH] Internal change

PiperOrigin-RevId: 549887345
---
 jmp/BUILD | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/jmp/BUILD b/jmp/BUILD
index 45c752a..000cac2 100644
--- a/jmp/BUILD
+++ b/jmp/BUILD
@@ -1,3 +1,5 @@
+load("//third_party/bazel_rules/rules_python/python:py_library.bzl", "py_library")
+
 # Description: JMP is a JAX Mixed Precision library.
 package(default_visibility = ["//visibility:private"])