flax
45a8f846 - Merge pull request #4592 from jakevdp:fix-shape

Commit

290 days ago

Merge pull request #4592 from jakevdp:fix-shape PiperOrigin-RevId: 733430128

References

#4592 - Avoid calls to jnp.shape for non-array inputs.

Author

a-googler

a-googler

Parents

Loading