Search results

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

questionUser queriesUser queries

Status: Open.

#1201 In patrick-kidger/equinox;

· jwtkeeble opened on Mar 24, 2026

RNN example breaks

Status: Open.

#1199 In patrick-kidger/equinox;

· josephcbradley opened on Mar 20, 2026

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

Status: Open.

#1196 In patrick-kidger/equinox;

· CatchemAL opened on Mar 8, 2026

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

Status: Open.

#1195 In patrick-kidger/equinox;

· CatchemAL opened on Mar 8, 2026

test_jax_transform_warn fails with jax 0.9

Status: Open.

#1186 In patrick-kidger/equinox;

· GaetanLepage opened on Feb 17, 2026

Custom error classes with error_if

Status: Open.

#1185 In patrick-kidger/equinox;

· f0uriest opened on Feb 17, 2026

How to diagnose/profile performance issues

Status: Open.

#1177 In patrick-kidger/equinox;

· toddpocuca opened on Feb 2, 2026

Correct use of custom filter_spec

questionUser queriesUser queries

Status: Open.

#1176 In patrick-kidger/equinox;

· tomsturges opened on Jan 28, 2026

test_sharding_* tests are failing with jax 0.9.0

Status: Open.

#1171 In patrick-kidger/equinox;

· GaetanLepage opened on Jan 24, 2026

XLA Convolution algorithms fail on GPU

questionUser queriesUser queries

Status: Open.

#1166 In patrick-kidger/equinox;

· rknetemann opened on Jan 13, 2026

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

questionUser queriesUser queries

Status: Open.

#1160 In patrick-kidger/equinox;

· JamesBrianD opened on Dec 24, 2025

Combining eqx.Partial with jax.lax.scan

questionUser queriesUser queries

Status: Open.

#1150 In patrick-kidger/equinox;

· mpvanderschelling opened on Nov 24, 2025