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.