Skip to content

Migrate from jax.tree_util legacy APIs to new jax.tree API. #4637

Migrate from jax.tree_util legacy APIs to new jax.tree API.

Migrate from jax.tree_util legacy APIs to new jax.tree API. #4637

Python 3.10 on ubuntu-latest jax=newest

succeeded Sep 20, 2024 in 12m 36s