Skip to content

Fix breaking changes introduced in JAX 0.4.36.#907

Merged
patrick-kidger merged 2 commits intomainfrom jax-breakage-fixesDec 8, 2024

Commits

Commits on Dec 7, 2024