Avoid passing concrete argument to jax.remat#5121
Merged
copybara-service[bot] merged 1 commit intomainfrom Dec 4, 2025
Merged
Avoid passing `concrete` argument to `jax.remat`#5121copybara-service[bot] merged 1 commit intomainfrom
copybara-service[bot] merged 1 commit intomainfrom
Commits
Commits on Dec 4, 2025
- authored andcommitted
