Multiple fixes and QoL changes

[sai-prasanna]

Sequence of commits with multiple QoL changes

1. Made it pip installable again. I had to move embodied back into dreamerv3 to keep the package integrity.
2. Allow options for wandb logger
3. Make the project compatible with latest gym and gymansium
4. Bug fix in config class where read won't work
5. Fix jax.jit=False, the current agent doesn't work when jit is off because we try to lower/compile few methods even when jit is off.
6. Allow passing extra args to environment. This is helpful for custom environments which is loaded using gymnasium or gym wrappers.
7. Allow logging based on steps instead of seconds
8. Allow saving intermediate checkpoints
9. Fix bug in using carry_report