v0.2.7

What's Changed

• Update broken Oryx github links to point to spun-off Oryx github repo. by @copybara-service in #67
• Add Layout support to jax.jit. by @copybara-service in #72
• Add 'cond_clobber' mode that lets different loop iterations sow different values to be reaped. by @copybara-service in #66
• Fixes monkey patching attribute error triggered by cl/629129623. by @copybara-service in #74
• Replace deprecated jax.tree_* functions with jax.tree.* by @copybara-service in #76

Full Changelog: v0.2.6...v0.2.7

v0.2.6

What's Changed

• Also do not patch MaskedAutoregressiveFlow. by @copybara-service in #63
• Bump oryx version. by @copybara-service in #64

Full Changelog: v0.2.5...v0.2.6

v0.2.5

Update requirements to allow TFP >= 0.20. Also removes unnecessary ty…

v0.2.3

Bump versions

PiperOrigin-RevId: 462748833