v2.0.0

We have switched the backend from Tensorflow to JAX due to flexibility and popularity.
The API bas also been simplified accordingly.
Please check the usage of the updated package in the latest version of Guide.ipynb and the training script.