Installing Anaconda:
When running python, you will often want to install different popular application packages. However, since each project may require a different set of applications with different versions, this can become a nightmare to manage.

Anaconda is a very useful application because it allows you to manage and switch between the environment for each project you are working on. Instructions for installing Anaconda for Windows, MacOS, or Linux are available from here.
If you are running JAX on a macOS, then make sure to install anaconda for the arm64 (i.e., M1, M2, M3, etc.) instruction set.

Also, here is a conda tutorial, and here is a cheat sheet for some common Anaconda commands.