Issues

• Reverse mode through a vmapped lineax.linear_solve fails when the RHS is unbatched but the operator is batched

  patrick-kidger/equinox#1274 · Filter by author djbower opened on Oct 19, 2026

• EquinoxRuntimeError is public but absent from objects.inv, so downstream Sphinx projects cannot reference it

  patrick-kidger/equinox#1265 · Filter by author nstarman opened on Sep 27, 2026
  1

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

  patrick-kidger/equinox#1257 · Filter by author nstarman opened on Aug 19, 2026

• AbstractVar is indistinguishable from a real field to type checkers, so it is counted in the synthesised __init__

  patrick-kidger/equinox#1256 · Filter by author nstarman opened on Aug 18, 2026
  1

• Equinox compatibility with external PyTreeNode

  patrick-kidger/equinox#1248 · Filter by author GJBoth opened on Jun 26, 2026
  3

• Instantiating Module inherited from abc.ABC returns __class__ assignment error on older versions

  patrick-kidger/equinox#1246 · Filter by author rabbott999 opened on Jun 22, 2026

• Best practices concerning referential transparency in equinox

  question
  patrick-kidger/equinox#1236 · Filter by author BenjaminDAnjou opened on May 20, 2026
  1

• equinox.error_if hangs on error if one of the arguments has a multi-device replicated sharding

  patrick-kidger/equinox#1232 · Filter by author Gattocrucco opened on May 19, 2026
  2

• Example of how to wrap a concrete subclass within the abstract/final design pattern

  question
  patrick-kidger/equinox#1231 · Filter by author BenjaminDAnjou opened on May 18, 2026
  5

• Attaching parameter metadata (bounds, priors) at instantiation using "make_with_meta" or "make_shadow"

  patrick-kidger/equinox#1224 · Filter by author gvcallen opened on Apr 22, 2026
  5

• FYI: Jax v0.9.2 made some nice fixes to TypedNdArray

  patrick-kidger/equinox#1206 · Filter by author nstarman opened on Apr 7, 2026
  3

• Is there a preferred way to compute the jacobian of a non-holomorphic function in equinox?

  question
  patrick-kidger/equinox#1201 · Filter by author jwtkeeble opened on Mar 24, 2026
  1

• nn.LayerNorm two-pass variance produces ~27% more HLO reduce ops under vmap + grad

  patrick-kidger/equinox#1196 · Filter by author CatchemAL opened on Mar 8, 2026
  4

• nn.Linear operand order causes unnecessary permutes under vmap + grad

  patrick-kidger/equinox#1195 · Filter by author CatchemAL opened on Mar 8, 2026
  4

• test_jax_transform_warn fails with jax 0.9

  patrick-kidger/equinox#1186 · Filter by author GaetanLepage opened on Feb 17, 2026
  3

• Custom error classes with error_if

  patrick-kidger/equinox#1185 · Filter by author f0uriest opened on Feb 17, 2026
  1

• How to diagnose/profile performance issues

  patrick-kidger/equinox#1177 · Filter by author toddpocuca opened on Feb 2, 2026

• Correct use of custom filter_spec

  question
  patrick-kidger/equinox#1176 · Filter by author tomsturges opened on Jan 28, 2026
  2

• test_sharding_* tests are failing with jax 0.9.0

  patrick-kidger/equinox#1171 · Filter by author GaetanLepage opened on Jan 24, 2026
  4

• XLA Convolution algorithms fail on GPU

  question
  patrick-kidger/equinox#1166 · Filter by author rknetemann opened on Jan 13, 2026
  8

• Is there a Slack/Discord (or similar) community for discussion?

  question
  patrick-kidger/equinox#1160 · Filter by author JamesBrianD opened on Dec 24, 2025
  3

• Combining eqx.Partial with jax.lax.scan

  question
  patrick-kidger/equinox#1150 · Filter by author mpvanderschelling opened on Nov 24, 2025
  3

• Equinox has spikes in training step durations due to garbage collector

  patrick-kidger/equinox#1145 · Filter by author AlexGraefe opened on Nov 12, 2025
  15

• Initialization of Layers compared to PYTORCH?

  question
  patrick-kidger/equinox#1143 · Filter by author ak24watch opened on Nov 7, 2025
  4

• error_if edge case

  question
  patrick-kidger/equinox#1142 · Filter by author alex-forster opened on Nov 6, 2025
  5