Update jax[cpu] requirement from <=0.4.36,>=0.3.2 to >=0.3.2,<=0.4.37

Update jax[cpu] requirement from <=0.4.36,>=0.3.2 to >=0.3.2,<=0.4.37 #34