Skip to content
build

Add random key support within a PyTree for McJAX #16084

Sign in to view logs

Add random key support within a PyTree for McJAX

Add random key support within a PyTree for McJAX #16084

Run time

Learn about OS pricing on GitHub Actions
Job Run time
build-export (Python 3.10, jax=newest)
1m 32s
build-checkpoint (Python 3.10, jax=newest)
1m 17s
build-export (Python 3.11, jax=newest)
1m 31s
build-orbax-model (Python 3.10, jax=newest)
9s
build-checkpoint (Python 3.11, jax=newest)
1m 11s
build-export (Python 3.12, jax=newest)
1m 33s
build-orbax-model (Python 3.11, jax=newest)
9s
build-checkpoint (Python 3.12, jax=newest)
1m 17s
build-export (Python 3.10, jax=0.4.34)
1m 41s
build-orbax-model (Python 3.12, jax=newest)
7s
build-checkpoint (Python 3.10, jax=0.4.34)
1m 23s
build-export (Python 3.12, jax=nightly)
1m 43s
build-orbax-model (Python 3.10, jax=0.4.34)
6s
build-checkpoint (Python 3.13, jax=nightly)
1m 27s
build-orbax-model (Python 3.12, jax=nightly)
8s
15m 14s