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

Annotations

2 warnings

Python 3.11 on ubuntu-latest jax=newest

succeeded Sep 20, 2024 in 12m 24s