Export LICENSE.txt file in jaxlib wheels.

build/BUILD.bazel

@@ -28,6 +28,8 @@ py_binary(
         "LICENSE.txt",
         "@org_tensorflow//tensorflow/compiler/xla/python:xla_client",
         "//jaxlib",
+        "//jaxlib:setup.py",
+        "//jaxlib:setup.cfg",
         "//jaxlib:lapack.so",
         "//jaxlib:_pocketfft.so",
         "//jaxlib:pocketfft_flatbuffers_py",

build/build_wheel.py

@@ -130,8 +130,10 @@ def prepare_wheel(sources_path):
   copy_to_jaxlib = functools.partial(copy_file, dst_dir=jaxlib_dir)
 
   verify_mac_libraries_dont_reference_chkstack()
-  copy_to_jaxlib(r.Rlocation("__main__/build/LICENSE.txt"))
+  copy_to_jaxlib(r.Rlocation("__main__/build/LICENSE.txt"),
+                 dst_dir=sources_path)
   copy_file(r.Rlocation("__main__/jaxlib/setup.py"), dst_dir=sources_path)
+  copy_file(r.Rlocation("__main__/jaxlib/setup.cfg"), dst_dir=sources_path)
   copy_to_jaxlib(r.Rlocation("__main__/jaxlib/init.py"),
                  dst_filename="__init__.py")
   copy_to_jaxlib(r.Rlocation("__main__/jaxlib/lapack.so"))

jaxlib/BUILD

@@ -80,12 +80,16 @@ py_library(
         "cusolver.py",
         "init.py",
         "pocketfft.py",
-        "setup.py",
         "version.py",
     ],
     deps = [":pocketfft_flatbuffers_py"],
 )
 
+exports_files([
+    "setup.py",
+    "setup.cfg",
+])
+
 py_library(
     name = "gpu_support",
     deps = [

jaxlib/setup.cfg

@@ -0,0 +1,2 @@
+[metadata]
+license_files = LICENSE.txt