Skip to content

improve checkpoint / remat concreteness error with static_argnums#24516

Merged
copybara-service[bot] merged 1 commit intojax-ml:mainfrom mattjj:improve-concreteness-error-in-remat-3Dec 18, 2024
+63-77

Commits