Skip to content

JAX release v0.2.28
Compare
Choose a tag to compare
@skye skye released this 02 Feb 01:55
· 10376 commits to main since this release
jax-v0.2.28
3acbd44
  • GitHub commits.
    • jax.jit(f).lower(...).compiler_ir() now defaults to the MHLO dialect if no
      dialect= is passed.
    • The jax.jit(f).lower(...).compiler_ir(dialect='mhlo') now returns an MLIR
      ir.Module object instead of its string representation.
Assets 2