Saransh Chopra
Saransh Chopra
The typing and rng issues need to be worked out ideally before merging this. Hence, marking as draft.
Thanks, @ntessore! `array.reshape()` exists in JAX and NumPy but it is not a part of the array API (`array_api_strict` tests fail).
This should finally be ready now, @paddyroddy @ntessore!
Merging this so that it does not go stale again!
Requesting a review to start discussions/get suggestions 🙂
pre-commit.ci autofix
The RTD failure looks unrelated.
Oops, the ids were actually not required. Everything looks nice now. See #595's [test run](https://github.com/glass-dev/glass/actions/runs/14780524414/job/41498249921?pr=595) for logs.
This is now blocked by #619
Thanks, @ianna! I’ll enable auto-merge on this.