patrick-kidger/equinox

View on GitHub

Elegant easy-to-use neural networks + scientific computing in JAX. https://docs.kidger.site/equinox/

Stars: 3k
Forks: 213
Open beginner issues: 0
Indexed issues: 232
Avg merge: 8d 4h
Merged PRs (30d): 2

Dominant language: Python
License: Apache-2.0
Last GitHub push: Sep 6, 2026
Latest indexed: Sep 20, 2026
Contributing guide: Contributing guide
Code of conduct: No code of conduct
Beginner labels: No beginner labels indexed

Beginner friendly Everything

Open All

232 open issues indexed

`error_if` allocates its error branch from a shape, dropping custom array types (quax) Open

Difficulty 4/5 3-5 days Newbie friendliness 68/100

patrick-kidger/equinox#1257 · Opened Aug 19, 2026
`AbstractVar` is indistinguishable from a real field to type checkers, so it is counted in the synthesised `__init__` Open

Difficulty 5/5 Over a week Newbie friendliness 35/100

patrick-kidger/equinox#1256 · 1 comment · Opened Aug 18, 2026
Serialization of rng keys Open

Difficulty 3/5 1-2 days Newbie friendliness 62/100

patrick-kidger/equinox#1255 · 4 comments · Opened Aug 3, 2026
Equinox compatibility with external `PyTreeNode` Open

Difficulty 4/5 3-5 days Newbie friendliness 48/100

patrick-kidger/equinox#1248 · 3 comments · Opened Jun 26, 2026
Instantiating Module inherited from abc.ABC returns __class__ assignment error on older versions Open

Difficulty 3/5 1-2 days Newbie friendliness 52/100

patrick-kidger/equinox#1246 · Opened Jun 22, 2026
Best practices concerning referential transparency in `equinox` Open

question

Difficulty 5/5 Over a week Newbie friendliness 35/100

patrick-kidger/equinox#1236 · 1 comment · Opened May 20, 2026
`equinox.error_if` hangs on error if one of the arguments has a multi-device replicated sharding Open

Difficulty 4/5 3-5 days Newbie friendliness 45/100

patrick-kidger/equinox#1232 · 2 comments · Opened May 19, 2026
Example of how to wrap a concrete subclass within the abstract/final design pattern Open

question

Difficulty 4/5 3-5 days Newbie friendliness 48/100

patrick-kidger/equinox#1231 · 5 comments · Opened May 18, 2026
Attaching parameter metadata (bounds, priors) at instantiation using "make_with_meta" or "make_shadow" Open

Difficulty 5/5 Over a week Newbie friendliness 35/100

patrick-kidger/equinox#1224 · 5 comments · Opened Apr 22, 2026
FYI: Jax v0.9.2 made some nice fixes to TypedNdArray Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

patrick-kidger/equinox#1206 · 3 comments · Opened Apr 7, 2026
Is there a preferred way to compute the jacobian of a non-holomorphic function in equinox? Open

question

Difficulty 5/5 Over a week Newbie friendliness 35/100

patrick-kidger/equinox#1201 · 1 comment · Opened Mar 24, 2026
nn.LayerNorm two-pass variance produces ~27% more HLO reduce ops under vmap + grad Open

Difficulty 2/5 1-3 hours Newbie friendliness 52/100

patrick-kidger/equinox#1196 · 4 comments · 1 reaction · Opened Mar 8, 2026
nn.Linear operand order causes unnecessary permutes under vmap + grad Open

Difficulty 2/5 1-3 hours Newbie friendliness 76/100

patrick-kidger/equinox#1195 · 4 comments · 1 reaction · Opened Mar 8, 2026
test_jax_transform_warn fails with jax 0.9 Open

Difficulty 3/5 1-2 days Newbie friendliness 38/100

patrick-kidger/equinox#1186 · 3 comments · Opened Feb 17, 2026
Custom error classes with `error_if` Open

Difficulty 4/5 3-5 days Newbie friendliness 45/100

patrick-kidger/equinox#1185 · 1 comment · Opened Feb 17, 2026
How to diagnose/profile performance issues Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

patrick-kidger/equinox#1177 · Opened Feb 2, 2026
Correct use of custom `filter_spec` Open

question

Difficulty 4/5 3-5 days Newbie friendliness 30/100

patrick-kidger/equinox#1176 · 2 comments · Opened Jan 28, 2026
`test_sharding_*` tests are failing with jax 0.9.0 Open

Difficulty 3/5 1-2 days Newbie friendliness 45/100

patrick-kidger/equinox#1171 · 4 comments · Opened Jan 24, 2026
XLA Convolution algorithms fail on GPU Open

question

Difficulty 4/5 3-5 days Newbie friendliness 38/100

patrick-kidger/equinox#1166 · 7 comments · Opened Jan 13, 2026
Is there a Slack/Discord (or similar) community for discussion? Open

question

Difficulty 1/5 Under an hour Newbie friendliness 25/100

patrick-kidger/equinox#1160 · 3 comments · Opened Dec 24, 2025
Combining eqx.Partial with jax.lax.scan Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/equinox#1150 · 3 comments · Opened Nov 24, 2025
Equinox has spikes in training step durations due to garbage collector Open

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/equinox#1145 · 15 comments · 2 reactions · Opened Nov 12, 2025
Initialization of Layers compared to PYTORCH? Open

question

Difficulty 2/5 1-3 hours Newbie friendliness 35/100

patrick-kidger/equinox#1143 · 4 comments · Opened Nov 7, 2025
error_if edge case Open

question

Difficulty 4/5 3-5 days Newbie friendliness 42/100

patrick-kidger/equinox#1142 · 5 comments · Opened Nov 6, 2025
eqx.filter_jit use on methods Open

question

Difficulty 3/5 1-2 days Newbie friendliness 35/100

patrick-kidger/equinox#1131 · 2 comments · Opened Oct 22, 2025
How to make a singleton? Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/equinox#1130 · 8 comments · Opened Oct 21, 2025
Control argument donation in `filter_jit` when first argument is `self` Open

question

Difficulty 4/5 3-5 days Newbie friendliness 38/100

patrick-kidger/equinox#1129 · 1 comment · Opened Oct 21, 2025
[Question] Is there a preferred way to visualise a network written as an eqx.Module? Open

question

Difficulty 5/5 Over a week Newbie friendliness 20/100

patrick-kidger/equinox#1128 · 2 comments · Opened Oct 21, 2025
Help understanding this behavior with pure jax jit compatibility Open

question

Difficulty 4/5 3-5 days Newbie friendliness 25/100

patrick-kidger/equinox#1127 · 1 comment · Opened Oct 21, 2025
AbstractVar implemented as property wrongly appears in default init signature Open

question

Difficulty 3/5 1-2 days Newbie friendliness 42/100

patrick-kidger/equinox#1123 · 1 comment · Opened Oct 19, 2025
Enumeration does not work with mypy Open

Difficulty 3/5 1-2 days Newbie friendliness 52/100

patrick-kidger/equinox#1121 · 4 comments · Opened Oct 17, 2025
Minor (?) test failure with JAX 0.8.0 - no warning emitted if JAX transformation set as attribute Open

Difficulty 3/5 1-2 days Newbie friendliness 42/100

patrick-kidger/equinox#1120 · 5 comments · Opened Oct 16, 2025
Inheritance with custom `__init__` with eqx.Module Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/equinox#1094 · 5 comments · Opened Sep 15, 2025
Can equinox be installed on jetson Xavier? Open

question

Difficulty 5/5 Over a week Newbie friendliness 20/100

patrick-kidger/equinox#1092 · 1 comment · Opened Sep 10, 2025
Catching the while loop regression in JAX 0.7.1 Open

Difficulty 4/5 3-5 days Newbie friendliness 25/100

patrick-kidger/equinox#1088 · 4 comments · Opened Aug 22, 2025
FYI: JAX 0.7.0 is known to be buggy with Equinox. Open

Difficulty 1/5 Under an hour Newbie friendliness 20/100

patrick-kidger/equinox#1084 · 1 comment · Opened Aug 21, 2025
Batching loss function over modules Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/equinox#1079 · 2 comments · Opened Aug 19, 2025
BatchNorm axis_name should be static? Open

Difficulty 2/5 1-3 hours Newbie friendliness 45/100

patrick-kidger/equinox#1074 · 4 comments · Opened Aug 13, 2025
Static Numpy arrays in Modules Open

question

Difficulty 5/5 Over a week Newbie friendliness 30/100

patrick-kidger/equinox#1073 · 3 comments · Opened Aug 13, 2025
Bug: state invalidation does not work across JIT boundaries Open

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/equinox#1071 · 4 comments · Opened Aug 13, 2025
Using class composition on abstract interfaces w/ methods and fields Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/equinox#1069 · 4 comments · Opened Aug 12, 2025
Eqx.nn Inference use on? Open

question

Difficulty 3/5 1-2 days Newbie friendliness 25/100

patrick-kidger/equinox#1064 · 1 comment · Opened Aug 3, 2025
Use of eqx.Partial in bnstasts Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/equinox#1063 · 3 comments · Opened Aug 3, 2025
`filter_jit` cache missing issues when using non-array variables with an `equinox` class Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/equinox#1062 · 1 comment · Opened Aug 1, 2025
eqx.filter breaks type annotations and post init checks Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/equinox#1049 · 3 comments · Opened Jul 18, 2025
Able to override concrete dunder methods Open

question

Difficulty 4/5 3-5 days Newbie friendliness 38/100

patrick-kidger/equinox#1042 · 2 comments · Opened Jul 7, 2025
Why does `common_rewrite` call `cond_fun` before `step < max_steps`? Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/equinox#1040 · 1 comment · Opened Jun 26, 2025
Equinox very nearly supports serialization/deserialization with pydantic. Open

Difficulty 5/5 Over a week Newbie friendliness 20/100

patrick-kidger/equinox#1037 · 1 comment · 2 reactions · Opened Jun 13, 2025
Understanding default field values when using a custom `__init__` Open

question

Difficulty 4/5 3-5 days Newbie friendliness 35/100

patrick-kidger/equinox#1035 · 2 comments · Opened Jun 13, 2025
Best practice for derived values of a dataclass Open

question

Difficulty 5/5 Over a week Newbie friendliness 25/100

patrick-kidger/equinox#1034 · 14 comments · Opened Jun 10, 2025

patrick-kidger/equinox

Get new issues in your inbox