Skip to content

Commit dfe27a1

Browse files
committed
Mention stackless in the release notes.
1 parent 1a3e693 commit dfe27a1

File tree

1 file changed

+15
-0
lines changed

1 file changed

+15
-0
lines changed

CHANGELOG.md

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,21 @@ When releasing, please add the new-release-boilerplate to docs/pallas/CHANGELOG.
1313
## jax 0.4.36
1414

1515
* Breaking Changes
16+
* This release lands "stackless", an internal change to JAX's tracing
17+
machinery. We made trace dispatch purely a function of context rather than a
18+
function of both context and data. This let us delete a lot of machinery for
19+
managing data-dependent tracing: levels, sublevels, `post_process_call`,
20+
`new_base_main`, `custom_bind`, and so on. The change should only affect
21+
users that use JAX internals.
22+
23+
If you do use JAX internals then you may need to
24+
update your code (see
25+
https://github.com/jax-ml/jax/commit/c36e1f7c1ad4782060cbc8e8c596d85dfb83986f
26+
for clues about how to do this). There might also be version skew
27+
issues with JAX libraries that do this. If you find this change breaks your
28+
non-JAX-internals-using code then try the
29+
`config.jax_data_dependent_tracing_fallback` flag as a workaround, and if
30+
you need help updating your code then please file a bug.
1631
* {func}`jax.experimental.jax2tf.convert` with `native_serialization=False`
1732
or with `enable_xla=False` have been deprecated since July 2024, with
1833
JAX version 0.4.31. Now we removed support for these use cases. `jax2tf`

0 commit comments

Comments
 (0)