Closed pnkraemer closed 2 months ago
https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.reshape.html
https://jax.readthedocs.io/en/latest/_autosummary/jax.numpy.reshape.html