Fix jax custom gradle (#21105)#21877

Open

Mayankvlog wants to merge 16 commits intokeras-team:masterfrom

Mayankvlog:fix-jax-custom-grad-21105

Commits on Oct 25, 2025

Document torch.compile limitations with EfficientNetV2 (Issue #21647)
Mayankvlog
committed

Commits on Oct 27, 2025

Commits on Oct 31, 2025

fix(tf-remat): avoid passing kwargs to custom_gradient in graph mode; add test
Mayankvlog
committed

Commits on Nov 2, 2025