To access this framework
Use the following commands in your environment

pip install jax jaxlib
matplotlib is optional but I recommend using it for the future codes.

thanks :)