In v0.5.1, the config class inherits from JAX. However, this means that installing without JAX will throw an error. This will need to be fixed in the next update.
This is fixed, although the solution involves maintaining a duplicate config class that will inevitably diverge from JAX, so we will need to revisit it.
In v0.5.1, the config class inherits from JAX. However, this means that installing without JAX will throw an error. This will need to be fixed in the next update.