diff --git a/docs/jax/index.html b/docs/jax/index.html
index db80085..b5423dd 100644
--- a/docs/jax/index.html
+++ b/docs/jax/index.html
@@ -311,5 +311,7 @@ <h1>Links for jax</h1>
 <a href="https://storage.googleapis.com/jax-releases/nightly/jax/jax-0.4.36.dev20241129.tar.gz" data-gpg-sig="false">nightly/jax/jax-0.4.36.dev20241129.tar.gz</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/jax/jax-0.4.36.dev20241130-py3-none-any.whl" data-requires-python="3" data-gpg-sig="false">nightly/jax/jax-0.4.36.dev20241130-py3-none-any.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/jax/jax-0.4.36.dev20241130.tar.gz" data-gpg-sig="false">nightly/jax/jax-0.4.36.dev20241130.tar.gz</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/jax/jax-0.4.36.dev20241201-py3-none-any.whl" data-requires-python="3" data-gpg-sig="false">nightly/jax/jax-0.4.36.dev20241201-py3-none-any.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/jax/jax-0.4.36.dev20241201.tar.gz" data-gpg-sig="false">nightly/jax/jax-0.4.36.dev20241201.tar.gz</a><br>
 </body>
 </html>
\ No newline at end of file
diff --git a/docs/jax_cuda12_pjrt/index.html b/docs/jax_cuda12_pjrt/index.html
index 7079402..94a88f2 100644
--- a/docs/jax_cuda12_pjrt/index.html
+++ b/docs/jax_cuda12_pjrt/index.html
@@ -250,6 +250,8 @@ <h1>Links for jax_cuda12_pjrt</h1>
 <a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jax_cuda12_pjrt-0.4.36.dev20241129-py3-none-manylinux2014_x86_64.whl" data-requires-python="3" data-gpg-sig="false">nightly/wheels/jax_cuda12_pjrt-0.4.36.dev20241129-py3-none-manylinux2014_x86_64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jax_cuda12_pjrt-0.4.36.dev20241130-py3-none-manylinux2014_aarch64.whl" data-requires-python="3" data-gpg-sig="false">nightly/wheels/jax_cuda12_pjrt-0.4.36.dev20241130-py3-none-manylinux2014_aarch64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jax_cuda12_pjrt-0.4.36.dev20241130-py3-none-manylinux2014_x86_64.whl" data-requires-python="3" data-gpg-sig="false">nightly/wheels/jax_cuda12_pjrt-0.4.36.dev20241130-py3-none-manylinux2014_x86_64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jax_cuda12_pjrt-0.4.36.dev20241201-py3-none-manylinux2014_aarch64.whl" data-requires-python="3" data-gpg-sig="false">nightly/wheels/jax_cuda12_pjrt-0.4.36.dev20241201-py3-none-manylinux2014_aarch64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jax_cuda12_pjrt-0.4.36.dev20241201-py3-none-manylinux2014_x86_64.whl" data-requires-python="3" data-gpg-sig="false">nightly/wheels/jax_cuda12_pjrt-0.4.36.dev20241201-py3-none-manylinux2014_x86_64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/temp_wheels/jax_cuda12_pjrt-0.4.24.dev20240118-py3-none-manylinux2014_x86_64.whl" data-requires-python="3" data-gpg-sig="false">temp_wheels/jax_cuda12_pjrt-0.4.24.dev20240118-py3-none-manylinux2014_x86_64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/temp_wheels/jax_cuda12_pjrt-0.4.25-py3-none-manylinux2014_aarch64.whl" data-requires-python="3" data-gpg-sig="false">temp_wheels/jax_cuda12_pjrt-0.4.25-py3-none-manylinux2014_aarch64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/temp_wheels/jax_cuda12_pjrt-0.4.26-py3-none-manylinux2014_aarch64.whl" data-requires-python="3" data-gpg-sig="false">temp_wheels/jax_cuda12_pjrt-0.4.26-py3-none-manylinux2014_aarch64.whl</a><br>
diff --git a/docs/jax_cuda12_plugin/index.html b/docs/jax_cuda12_plugin/index.html
index 67a730d..7ed1bd0 100644
--- a/docs/jax_cuda12_plugin/index.html
+++ b/docs/jax_cuda12_plugin/index.html
@@ -907,6 +907,14 @@ <h1>Links for jax_cuda12_plugin</h1>
 <a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241130-cp312-cp312-manylinux2014_x86_64.whl" data-requires-python="3.12" data-gpg-sig="false">nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241130-cp312-cp312-manylinux2014_x86_64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241130-cp313-cp313-manylinux2014_aarch64.whl" data-requires-python="3.13" data-gpg-sig="false">nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241130-cp313-cp313-manylinux2014_aarch64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241130-cp313-cp313-manylinux2014_x86_64.whl" data-requires-python="3.13" data-gpg-sig="false">nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241130-cp313-cp313-manylinux2014_x86_64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241201-cp310-cp310-manylinux2014_aarch64.whl" data-requires-python="3.10" data-gpg-sig="false">nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241201-cp310-cp310-manylinux2014_aarch64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241201-cp310-cp310-manylinux2014_x86_64.whl" data-requires-python="3.10" data-gpg-sig="false">nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241201-cp310-cp310-manylinux2014_x86_64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241201-cp311-cp311-manylinux2014_aarch64.whl" data-requires-python="3.11" data-gpg-sig="false">nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241201-cp311-cp311-manylinux2014_aarch64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241201-cp311-cp311-manylinux2014_x86_64.whl" data-requires-python="3.11" data-gpg-sig="false">nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241201-cp311-cp311-manylinux2014_x86_64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241201-cp312-cp312-manylinux2014_aarch64.whl" data-requires-python="3.12" data-gpg-sig="false">nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241201-cp312-cp312-manylinux2014_aarch64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241201-cp312-cp312-manylinux2014_x86_64.whl" data-requires-python="3.12" data-gpg-sig="false">nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241201-cp312-cp312-manylinux2014_x86_64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241201-cp313-cp313-manylinux2014_aarch64.whl" data-requires-python="3.13" data-gpg-sig="false">nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241201-cp313-cp313-manylinux2014_aarch64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241201-cp313-cp313-manylinux2014_x86_64.whl" data-requires-python="3.13" data-gpg-sig="false">nightly/wheels/jax_cuda12_plugin-0.4.36.dev20241201-cp313-cp313-manylinux2014_x86_64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/temp_wheels/jax_cuda12_plugin-0.4.24.dev20240118-cp310-cp310-manylinux2014_x86_64.whl" data-requires-python="3.10" data-gpg-sig="false">temp_wheels/jax_cuda12_plugin-0.4.24.dev20240118-cp310-cp310-manylinux2014_x86_64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/temp_wheels/jax_cuda12_plugin-0.4.24.dev20240118-cp311-cp311-manylinux2014_x86_64.whl" data-requires-python="3.11" data-gpg-sig="false">temp_wheels/jax_cuda12_plugin-0.4.24.dev20240118-cp311-cp311-manylinux2014_x86_64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/temp_wheels/jax_cuda12_plugin-0.4.24.dev20240118-cp312-cp312-manylinux2014_x86_64.whl" data-requires-python="3.12" data-gpg-sig="false">temp_wheels/jax_cuda12_plugin-0.4.24.dev20240118-cp312-cp312-manylinux2014_x86_64.whl</a><br>
diff --git a/docs/jaxlib/index.html b/docs/jaxlib/index.html
index 13b3b37..e85c291 100644
--- a/docs/jaxlib/index.html
+++ b/docs/jaxlib/index.html
@@ -2997,6 +2997,10 @@ <h1>Links for jaxlib</h1>
 <a href="https://storage.googleapis.com/jax-releases/nightly/mac/jaxlib-0.4.36.dev20241130-cp311-cp311-macosx_11_0_arm64.whl" data-requires-python="3.11" data-gpg-sig="false">nightly/mac/jaxlib-0.4.36.dev20241130-cp311-cp311-macosx_11_0_arm64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/mac/jaxlib-0.4.36.dev20241130-cp312-cp312-macosx_11_0_arm64.whl" data-requires-python="3.12" data-gpg-sig="false">nightly/mac/jaxlib-0.4.36.dev20241130-cp312-cp312-macosx_11_0_arm64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/mac/jaxlib-0.4.36.dev20241130-cp313-cp313-macosx_11_0_arm64.whl" data-requires-python="3.13" data-gpg-sig="false">nightly/mac/jaxlib-0.4.36.dev20241130-cp313-cp313-macosx_11_0_arm64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/mac/jaxlib-0.4.36.dev20241201-cp310-cp310-macosx_11_0_arm64.whl" data-requires-python="3.10" data-gpg-sig="false">nightly/mac/jaxlib-0.4.36.dev20241201-cp310-cp310-macosx_11_0_arm64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/mac/jaxlib-0.4.36.dev20241201-cp311-cp311-macosx_11_0_arm64.whl" data-requires-python="3.11" data-gpg-sig="false">nightly/mac/jaxlib-0.4.36.dev20241201-cp311-cp311-macosx_11_0_arm64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/mac/jaxlib-0.4.36.dev20241201-cp312-cp312-macosx_11_0_arm64.whl" data-requires-python="3.12" data-gpg-sig="false">nightly/mac/jaxlib-0.4.36.dev20241201-cp312-cp312-macosx_11_0_arm64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/mac/jaxlib-0.4.36.dev20241201-cp313-cp313-macosx_11_0_arm64.whl" data-requires-python="3.13" data-gpg-sig="false">nightly/mac/jaxlib-0.4.36.dev20241201-cp313-cp313-macosx_11_0_arm64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/nocuda/jaxlib-0.4.32.dev20240807-cp310-cp310-manylinux2014_aarch64.whl" data-requires-python="3.10" data-gpg-sig="false">nightly/nocuda/jaxlib-0.4.32.dev20240807-cp310-cp310-manylinux2014_aarch64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/nocuda/jaxlib-0.4.32.dev20240807-cp310-cp310-manylinux2014_x86_64.whl" data-requires-python="3.10" data-gpg-sig="false">nightly/nocuda/jaxlib-0.4.32.dev20240807-cp310-cp310-manylinux2014_x86_64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/nocuda/jaxlib-0.4.32.dev20240807-cp311-cp311-manylinux2014_aarch64.whl" data-requires-python="3.11" data-gpg-sig="false">nightly/nocuda/jaxlib-0.4.32.dev20240807-cp311-cp311-manylinux2014_aarch64.whl</a><br>
@@ -3778,6 +3782,14 @@ <h1>Links for jaxlib</h1>
 <a href="https://storage.googleapis.com/jax-releases/nightly/nocuda/jaxlib-0.4.36.dev20241130-cp312-cp312-manylinux2014_x86_64.whl" data-requires-python="3.12" data-gpg-sig="false">nightly/nocuda/jaxlib-0.4.36.dev20241130-cp312-cp312-manylinux2014_x86_64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/nocuda/jaxlib-0.4.36.dev20241130-cp313-cp313-manylinux2014_aarch64.whl" data-requires-python="3.13" data-gpg-sig="false">nightly/nocuda/jaxlib-0.4.36.dev20241130-cp313-cp313-manylinux2014_aarch64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/nocuda/jaxlib-0.4.36.dev20241130-cp313-cp313-manylinux2014_x86_64.whl" data-requires-python="3.13" data-gpg-sig="false">nightly/nocuda/jaxlib-0.4.36.dev20241130-cp313-cp313-manylinux2014_x86_64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/nocuda/jaxlib-0.4.36.dev20241201-cp310-cp310-manylinux2014_aarch64.whl" data-requires-python="3.10" data-gpg-sig="false">nightly/nocuda/jaxlib-0.4.36.dev20241201-cp310-cp310-manylinux2014_aarch64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/nocuda/jaxlib-0.4.36.dev20241201-cp310-cp310-manylinux2014_x86_64.whl" data-requires-python="3.10" data-gpg-sig="false">nightly/nocuda/jaxlib-0.4.36.dev20241201-cp310-cp310-manylinux2014_x86_64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/nocuda/jaxlib-0.4.36.dev20241201-cp311-cp311-manylinux2014_aarch64.whl" data-requires-python="3.11" data-gpg-sig="false">nightly/nocuda/jaxlib-0.4.36.dev20241201-cp311-cp311-manylinux2014_aarch64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/nocuda/jaxlib-0.4.36.dev20241201-cp311-cp311-manylinux2014_x86_64.whl" data-requires-python="3.11" data-gpg-sig="false">nightly/nocuda/jaxlib-0.4.36.dev20241201-cp311-cp311-manylinux2014_x86_64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/nocuda/jaxlib-0.4.36.dev20241201-cp312-cp312-manylinux2014_aarch64.whl" data-requires-python="3.12" data-gpg-sig="false">nightly/nocuda/jaxlib-0.4.36.dev20241201-cp312-cp312-manylinux2014_aarch64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/nocuda/jaxlib-0.4.36.dev20241201-cp312-cp312-manylinux2014_x86_64.whl" data-requires-python="3.12" data-gpg-sig="false">nightly/nocuda/jaxlib-0.4.36.dev20241201-cp312-cp312-manylinux2014_x86_64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/nocuda/jaxlib-0.4.36.dev20241201-cp313-cp313-manylinux2014_aarch64.whl" data-requires-python="3.13" data-gpg-sig="false">nightly/nocuda/jaxlib-0.4.36.dev20241201-cp313-cp313-manylinux2014_aarch64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/nocuda/jaxlib-0.4.36.dev20241201-cp313-cp313-manylinux2014_x86_64.whl" data-requires-python="3.13" data-gpg-sig="false">nightly/nocuda/jaxlib-0.4.36.dev20241201-cp313-cp313-manylinux2014_x86_64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jaxlib-0.4.22.dev20231205-cp310-cp310-manylinux2014_x86_64.whl" data-requires-python="3.10" data-gpg-sig="false">nightly/wheels/jaxlib-0.4.22.dev20231205-cp310-cp310-manylinux2014_x86_64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jaxlib-0.4.22.dev20231205-cp39-cp39-manylinux2014_x86_64.whl" data-requires-python="3.9" data-gpg-sig="false">nightly/wheels/jaxlib-0.4.22.dev20231205-cp39-cp39-manylinux2014_x86_64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jaxlib-0.4.22.dev20231206-cp310-cp310-manylinux2014_x86_64.whl" data-requires-python="3.10" data-gpg-sig="false">nightly/wheels/jaxlib-0.4.22.dev20231206-cp310-cp310-manylinux2014_x86_64.whl</a><br>
@@ -5100,6 +5112,10 @@ <h1>Links for jaxlib</h1>
 <a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jaxlib-0.4.36.dev20241130-cp311-cp311-manylinux2014_x86_64.whl" data-requires-python="3.11" data-gpg-sig="false">nightly/wheels/jaxlib-0.4.36.dev20241130-cp311-cp311-manylinux2014_x86_64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jaxlib-0.4.36.dev20241130-cp312-cp312-manylinux2014_x86_64.whl" data-requires-python="3.12" data-gpg-sig="false">nightly/wheels/jaxlib-0.4.36.dev20241130-cp312-cp312-manylinux2014_x86_64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jaxlib-0.4.36.dev20241130-cp313-cp313-manylinux2014_x86_64.whl" data-requires-python="3.13" data-gpg-sig="false">nightly/wheels/jaxlib-0.4.36.dev20241130-cp313-cp313-manylinux2014_x86_64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jaxlib-0.4.36.dev20241201-cp310-cp310-manylinux2014_x86_64.whl" data-requires-python="3.10" data-gpg-sig="false">nightly/wheels/jaxlib-0.4.36.dev20241201-cp310-cp310-manylinux2014_x86_64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jaxlib-0.4.36.dev20241201-cp311-cp311-manylinux2014_x86_64.whl" data-requires-python="3.11" data-gpg-sig="false">nightly/wheels/jaxlib-0.4.36.dev20241201-cp311-cp311-manylinux2014_x86_64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jaxlib-0.4.36.dev20241201-cp312-cp312-manylinux2014_x86_64.whl" data-requires-python="3.12" data-gpg-sig="false">nightly/wheels/jaxlib-0.4.36.dev20241201-cp312-cp312-manylinux2014_x86_64.whl</a><br>
+<a href="https://storage.googleapis.com/jax-releases/nightly/wheels/jaxlib-0.4.36.dev20241201-cp313-cp313-manylinux2014_x86_64.whl" data-requires-python="3.13" data-gpg-sig="false">nightly/wheels/jaxlib-0.4.36.dev20241201-cp313-cp313-manylinux2014_x86_64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nocuda/jaxlib-0.1.23-cp27-none-manylinux1_x86_64.whl" data-requires-python="2.7" data-gpg-sig="false">nocuda/jaxlib-0.1.23-cp27-none-manylinux1_x86_64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nocuda/jaxlib-0.1.23-cp35-none-manylinux1_x86_64.whl" data-requires-python="3.5" data-gpg-sig="false">nocuda/jaxlib-0.1.23-cp35-none-manylinux1_x86_64.whl</a><br>
 <a href="https://storage.googleapis.com/jax-releases/nocuda/jaxlib-0.1.23-cp36-none-manylinux1_x86_64.whl" data-requires-python="3.6" data-gpg-sig="false">nocuda/jaxlib-0.1.23-cp36-none-manylinux1_x86_64.whl</a><br>