$ the-wire · showcase
JAX fixes arg_exclude_mask initialization and updates XLA dependency
By RepoJournal · Filed · About Google
JAX corrects arg_exclude_mask value initialization when exclude_argnames is None and moves to a new XLA revision.
JAX fixed the initialization of arg_exclude_mask when exclude_argnames is None, ensuring the mask is set with the correct values in that case [1]. The same change appears in a separate commit [2], both addressing the same behavior for callers who omit exclude_argnames. Separately, the XLA dependency now points to revision 496bd4bd49db9ecbffd85da630b49c860b724604 [3], pulling in the latest XLA changes with the next JAX build. The fix matters for code that relies on the default masking behavior, since the mask previously may have held incorrect values when exclude_argnames was not supplied. No breaking changes accompany either update.
Action items
References
- [1] Initialize arg_exclude_mask with the correct values if exclude_argnames is None ↗ google/jax
- [2] Initialize arg_exclude_mask with the correct values if exclude_argnames is None ↗ google/jax
- [3] Update XLA dependency to use revision http://github.com/openxla/xla/commit/496bd4bd49db9ecbffd85da630b49c860b724604 ↗ google/jax