Issues with the diffuser requirement

[aagha6]

The diffuser version required atm is 0.11.1, which is very outdated and pins jax to 0.4.1, which makes Robomimic conflict with newer JAX versions. I think this can be solved by merging #263