fig,ax = plt.subplots(1,1)
ax.plot(w.t, toy_actions[0].to(u.km/u.s*u.kpc), marker='')
ax.set_xlabel(r"$t$ [Myr]")
ax.set_ylabel(r"$J_1$ [kpc km/s]")
fig.tight_layout()