1e2e4bc introduced version qualifier numpy < 1.23 due to failing tests with jax, but I believe this has been resolved with recent jax versions and we no longer need < 1.23.

I could write a trivial PR that fixes requirements.txt, but let me leave it to be handled by the rlax team because the commit does not specify at which jax version tests are failing, hopefully the CL can update jax and numpy version requirements.

/cc @katebaumli