Skip to content

Reduce NonSequential JIT compile times via jax.lax.scan and optimizer state packing#5

Draft
Copilot wants to merge 2 commits into
mainfrom
copilot/reduce-nonsequential-compile-time
Draft

Reduce NonSequential JIT compile times via jax.lax.scan and optimizer state packing#5
Copilot wants to merge 2 commits into
mainfrom
copilot/reduce-nonsequential-compile-time

Reduce NonSequential compile times via jax.lax.scan and optimizer pac…

1f5eb52
Select commit
Loading
Failed to load commit list.

There are no checks for this commit