Skip to content

Pull requests: patrick-kidger/equinox

New pull request New
19 Open 565 Closed
19 Open 565 Closed
Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Assigned to nobody Loading
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Treat JAX and NumPy arrays as interchangeable when deserialising
#1253 opened Jul 27, 2026 by JOhnsonKC201 Loading…
1
perf: short-circuit Module.__getattribute__ on non-method access
#1252 opened Jul 21, 2026 by nstarman Contributor Loading…
Don't leak partially-constructed modules when __init__ raises
#1251 opened Jul 20, 2026 by nstarman Contributor Loading…
3
Add adv-jax-math to awesome list
#1250 opened Jul 20, 2026 by unalmis Loading…
1
2
Add a paragraph about how to use jaxon for serialisation to the documentation.
#1249 opened Jul 18, 2026 by 42Ar Loading…
2
Fix eager error_if hanging when inputs are sharded over multiple devices
#1242 opened Jun 13, 2026 by discobot Contributor Loading…
1
2
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…
1
5
perf: add fastpaths in _ModuleMeta.__call__
#1230 opened May 14, 2026 by nstarman Contributor Loading…
4
Reformat dependencies in pyproject.toml
#1194 opened Mar 7, 2026 by dhirschfeld Loading…
1
Use @jtu.register_static as decorator on _Missing
#1193 opened Mar 4, 2026 by nstarman Contributor • Draft
6
Enable eqx.nn.State to track and return Intermediate values
#1165 opened Jan 10, 2026 by arenaudineau Loading…
Fixes for mypy + Enumeration
#1122 opened Oct 17, 2025 by patrick-kidger Owner Loading…
perf: enable module flatten/unflatten fastpath
#1117 opened Oct 12, 2025 by nstarman Contributor • Draft
12
Add memorax link to readme
#873 opened Oct 8, 2024 by smorad Loading…
1
Added RoPE Offset + Test
#860 opened Sep 23, 2024 by Artur-Galstyan Contributor Loading…
Add exact spectral norm feature
#831 opened Sep 6, 2024 by lockwo Contributor Loading…
8
BatchNorm training instability fix
#675 opened Mar 7, 2024 by andrewdipper Loading…
21
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! What’s not been updated in a month: updated:<2026-06-30.