To generate figure 1, simply type "python burgers.py". You will need to have Jax installed.