maxtext
maxtext copied to clipboard
Replace deprecated np.product with np.prod
We are seeing
DeprecationWarning: `product` is deprecated as of NumPy 1.25.0, and will be removed in NumPy 2.0. Please use `prod` instead.