fix(math): ShardedArray pytree flatten + remove_diag guard#839
Merged
Sourcery AI / Sourcery review
succeeded
Jun 18, 2026 in 1m 47s
✅ Completed successfully
Sourcery completed the review and found no blocking security issues
Loading