Dropped stale jaxlib conditional

JAX requires jaxlib >= 0.4.34.
jax-ml · Oct 21, 2024 · a6d1b67 · a6d1b67
1 parent a72a159
commit a6d1b67
Showing 1 changed file with 2 additions and 8 deletions.
diff --git a/jax_triton/__init__.py b/jax_triton/__init__.py
@@ -35,12 +35,6 @@
 from jax_triton.version import __version_info__
 
 get_compute_capability = gpu_triton.get_compute_capability
-if jaxlib.version.__version_info__ >= (0, 4, 14):
-  try:
-    get_serialized_metadata = gpu_triton.get_serialized_metadata
-  except AttributeError:
-    get_serialized_metadata = None
+get_serialized_metadata = gpu_triton.get_serialized_metadata
 
-# trailer
-del gpu_triton
-del jaxlib
+del gpu_triton  # Not part of the API.