Gahow Wang 0acfa5df11 ops: grad-check the T5 structural ops ...

Finite-diff grad-checks (same L=sum(W∘out) harness as autograd.rs) for
embedding (incl. repeated ids), reshape, transpose_3d01, transpose_2d,
and split/merge_heads round-trip. Gated #![cfg(not(no_cuda))].

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

2026-06-15 16:05:20 +08:00

..

autograd.rs

ops: differentiable autograd nodes + per-op grad-check tests

2026-06-15 15:53:55 +08:00

structural.rs

ops: grad-check the T5 structural ops

2026-06-15 16:05:20 +08:00