Runtime error messages (those from eqx.error_if, in particular when wrapped with eqx.filter_jit) should now be compatible with PyCharm's debugger, and with certain multithreaded contexts. (Thanks @adam-hartshorne, @dlwh! #828, #844, #849)
Marking a jax.Array or np.ndarray as an eqx.field(static=True) will now raise a warning. This was technically okay as long as you use it in certain very narrow contexts (e.g. to smuggle it into a JIT'd region without being traced), but in practice it was nearly always just a common new-user footgun. (Thanks @lockwo! #800)
Using eqx.tree_at for replacing empty tuples is improved. (Thanks @danielward27! #818, #819)
eqx.nn.RotaryEmbedding no longer promote input dtypes to at least float32. (Thanks @knyazer! #836)
Mypy now understands that eqx.Modules are dataclasses. (Pyright always did, but mypy needed a slightly different approach to appreciate this fact.) (Thanks @NeilGirdhar! #822)
Multiple eqx.Modules participating in co-operative multiple inheritance (at least 5 inheriting from each other seem to be necessary?), with some of them overriding the __post_init__s of others, should now follow their expected resolution order. (Thanks @NeilGirdhar! #832, #834)
Improvements for contributors: we now have a .editorconfig file, (thanks @NeilGirdhar! #821) and test-time dependencies can now be installed as pip install equinox[tests] (thanks @dlwh! #848)
This is primarily a bug fix release.
Runtime error messages (those from
eqx.error_if, in particular when wrapped witheqx.filter_jit) should now be compatible with PyCharm's debugger, and with certain multithreaded contexts. (Thanks @adam-hartshorne, @dlwh! #828, #844, #849)Marking a
jax.Arrayornp.ndarrayas aneqx.field(static=True)will now raise a warning. This was technically okay as long as you use it in certain very narrow contexts (e.g. to smuggle it into a JIT'd region without being traced), but in practice it was nearly always just a common new-user footgun. (Thanks @lockwo! #800)Using
eqx.tree_atfor replacing empty tuples is improved. (Thanks @danielward27! #818, #819)eqx.nn.RotaryEmbeddingno longer promote input dtypes to at least float32. (Thanks @knyazer! #836)Mypy now understands that
eqx.Modules are dataclasses. (Pyright always did, but mypy needed a slightly different approach to appreciate this fact.) (Thanks @NeilGirdhar! #822)Multiple
eqx.Modules participating in co-operative multiple inheritance (at least 5 inheriting from each other seem to be necessary?), with some of them overriding the__post_init__s of others, should now follow their expected resolution order. (Thanks @NeilGirdhar! #832, #834)Improvements for contributors: we now have a
.editorconfigfile, (thanks @NeilGirdhar! #821) and test-time dependencies can now be installed aspip install equinox[tests](thanks @dlwh! #848)Doc improvements. (Thanks @garymm, @ColCarroll! #804, #805)