Skip to content
Back to pull request #26814

Reimplement custom_vjp.optimize_remat using custom_dce #4927

Sign in to view logs

Reimplement custom_vjp.optimize_remat using custom_dce

Reimplement custom_vjp.optimize_remat using custom_dce #4927

TPU test (jaxlib=head, v5e-8)

TPU test (jaxlib=head, v5e-8)
succeeded Feb 28, 2025 in 8m 9s