Skip to content

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

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

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

Annotations

2 warnings

Python 3.10 on ubuntu-latest jax=newest

succeeded Sep 20, 2024 in 12m 20s