Search results

Nx.LinAlg gradients don't support batched (3D+) inputs

Status: Open.

#1748 In elixir-nx/nx;

· blasphemetheus opened on Apr 21, 2026

Nx.Defn.while gradient incorrect when loop body's Jacobian w.r.t. accumulator depends on differentiated variable

Status: Open.

#1747 In elixir-nx/nx;

· blasphemetheus opened on Apr 21, 2026

Nx.LinAlg.eigh gradient broken for 2D inputs and for all f64 inputs

Status: Open.

#1740 In elixir-nx/nx;

· blasphemetheus opened on Apr 21, 2026

sharding_test 2D mesh assertion has wrong expected shapes (exposed by #1733)

Status: Open.

#1734 In elixir-nx/nx;

· blasphemetheus opened on Apr 14, 2026

Vectorized gradient fails when input and output vec axis of the same name have different sizes

Status: Open.

#1730 In elixir-nx/nx;

· blasphemetheus opened on Apr 14, 2026

Nx.conv shortcomings (vs numpy.conv)

Status: Open.

#1708 In elixir-nx/nx;

· laszlokorte opened on Mar 18, 2026

Hook messages within 'while' arrive out of order on OTP 28.3

Status: Open.

#1686 In elixir-nx/nx;

· blasphemetheus opened on Mar 11, 2026

Import of MLIR modules

area:exlaApplies to EXLAApplies to EXLA

kind:featureNew feature or requestNew feature or request

Status: Open.

#1464 In elixir-nx/nx;

· josevalim opened on Mar 5, 2024

Support checkpoints in gradients

area:nxApplies to nxApplies to nx

kind:featureNew feature or requestNew feature or request

Status: Open.

#765 In elixir-nx/nx;

· josevalim opened on Apr 28, 2022