search

jax-ml / jax-triton

jax-triton contains integrations between JAX and OpenAI Triton
Apache License 2.0
354 stars 40 forks source link

Assume that `jaxlib.get_serialized_metadata` is always defined #305

Closed superbobry closed 1 month ago

superbobry commented 1 month ago

JAX now requires jaxlib >= 0.4.34, so we should as well.