Skip to content
ci

[JAX] Replace uses of jax._src.ad_checkpoint.optimization_barrier with jax.lax.optimization_barrier. #1618

Sign in to view logs

[JAX] Replace uses of jax._src.ad_checkpoint.optimization_barrier with jax.lax.optimization_barrier.

[JAX] Replace uses of jax._src.ad_checkpoint.optimization_barrier with jax.lax.optimization_barrier. #1618

Run time

Learn about OS pricing on GitHub Actions
Job Run time
Python 3.10 on ubuntu-latest
3m 34s
Python 3.11 on ubuntu-latest
3m 41s
7m 15s