-
-
Notifications
You must be signed in to change notification settings - Fork 210
Pull requests: patrick-kidger/equinox
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
Don't leak partially-constructed modules when
__init__ raises
#1251
opened Jul 20, 2026 by
nstarman
Contributor
Loading…
Fix eager
error_if hanging when inputs are sharded over multiple devices
#1242
opened Jun 13, 2026 by
discobot
Contributor
Loading…
Switch
eqx.nn.Linear.__call__ to x @ weight.T to eliminate vmap+grad permutes (gated on GPU benchmark)
#1235
opened May 19, 2026 by
mvanhorn
Contributor
Loading…
perf: add fastpaths in
_ModuleMeta.__call__
#1230
opened May 14, 2026 by
nstarman
Contributor
Loading…
Enable eqx.nn.State to track and return Intermediate values
#1165
opened Jan 10, 2026 by
arenaudineau
Loading…
Now running tree_check when initialising a Module
#478
opened Sep 7, 2023 by
patrick-kidger
Owner
Loading…
First pass at making error_if work with sharding.
#342
opened May 8, 2023 by
patrick-kidger
Owner
•
Draft
ProTip!
Filter pull requests by the default branch with base:main.