Offlax: Offline Reinforcement Learning in JAX

Under development during my free time. Taking a botom to top approach in designing various components of the framework.

Plan to Release 0.2

1. File I/O for large-datasets using petastorm
2. Supporting public datasets(will be released in a few days)
3. Implementing SOTA Online algorithms(SAC, PPO)(will be released in a few days)
4. Implement CQL, IQL