Add batch_shape argument to optax.tree_utils.tree_random_like. #5108
Annotations
1 warning
|
ubuntu-latest pipelines will use ubuntu-24.04 soon. For more details, see https://github.com/actions/runner-images/issues/10636
|
Loading