jax
91280e51 - Fix pallas fuser BlockIndexTransform comparison to consider symbolic equivalence of block indices transforms rather than just object identity.

Commit
13 days ago
Fix pallas fuser BlockIndexTransform comparison to consider symbolic equivalence of block indices transforms rather than just object identity. PiperOrigin-RevId: 912189570
Author
Parents
Loading