JAX release v0.2.28

skye released this 02 Feb 01:55

· 21722 commits to main since this release

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