jax icon indicating copy to clipboard operation
jax copied to clipboard
verified publisher icon google

Clean up conditional lowering of tan after JAX v0.4.34 release.

Open dfm opened this issue 1 year ago • 0 comments

In https://github.com/jax-ml/jax/pull/23261, the lowering of lax.tan was updated to support the native StableHLO lowering, but jaxlib v0.4.33 didn't include the necessary re-writes to support this lowering via the PJRT plugin so we needed a conditional on the jaxlib version (https://github.com/jax-ml/jax/pull/23762). Now that v0.4.34 is out, this shouldn't be needed any longer.

dfm avatar Oct 19 '24 20:10 dfm