diff --git a/setup.py b/setup.py
index 7fa4bdef2..54b9e9c46 100644
--- a/setup.py
+++ b/setup.py
@@ -1,4 +1,5 @@
 from setuptools import setup
+from glob import glob
 
 setup(
     name='jax',
@@ -11,4 +12,5 @@ setup(
     install_requires=['numpy>=1.12', 'six', 'protobuf'],
     url='https://github.com/google/jax',
     license='Apache-2.0',
+    package_data={'jax.lib': glob('jax/lib/*.so')},
 )