Skip to content

Fix warning of the form DeprecationWarning: jax.core.Shape is deprecated. Use Shape = Sequence[int | Any] #1658

Fix warning of the form DeprecationWarning: jax.core.Shape is deprecated. Use Shape = Sequence[int | Any]

Fix warning of the form DeprecationWarning: jax.core.Shape is deprecated. Use Shape = Sequence[int | Any] #1658